lightx2v

/

Wan2.1-Distill-Models

like 17

Hi,

I get this error:

  File "/workspace/LightX2V/lightx2v/infer.py", line 115, in <module>
    main()
  File "/workspace/LightX2V/lightx2v/infer.py", line 104, in main
    runner = init_runner(config)
             ^^^^^^^^^^^^^^^^^^^
  File "/workspace/LightX2V/lightx2v/infer.py", line 26, in init_runner
    runner.init_modules()
  File "/workspace/LightX2V/lightx2v/models/runners/default_runner.py", line 39, in init_modules
    self.load_model()
  File "/workspace/LightX2V/lightx2v/utils/profiler.py", line 77, in sync_wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
  File "/workspace/LightX2V/lightx2v/models/runners/default_runner.py", line 86, in load_model
    self.model = self.load_transformer()
                 ^^^^^^^^^^^^^^^^^^^^^^^
  File "/workspace/LightX2V/lightx2v/models/runners/wan/wan_distill_runner.py", line 34, in load_transformer
    model = WanDistillModel(self.config["model_path"], self.config, self.init_device)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/workspace/LightX2V/lightx2v/models/networks/wan/distill_model.py", line 22, in __init__
    super().__init__(model_path, config, device, model_type)
  File "/workspace/LightX2V/lightx2v/models/networks/wan/model.py", line 66, in __init__
    self._init_weights()
  File "/workspace/LightX2V/lightx2v/models/networks/wan/model.py", line 265, in _init_weights
    self.pre_weight = self.pre_weight_class(self.config)
                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/workspace/LightX2V/lightx2v/models/networks/wan/weights/pre_weights.py", line 13, in __init__
    self.in_dim = config["in_dim"]
                  ~~~~~~^^^^^^^^^^
KeyError: 'in_dim'