Skip to content

GitLab

Unverified Commit 58c9abb9 authored by Gu Shiqiao's avatar Gu Shiqiao Committed by GitHub
Browse files

Move _load_adapter_ckpt call in constructor (#509)

parent 4f829742
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lightx2v/models/networks/wan/audio_model.py
View file @ 58c9abb9
......@@ -22,8 +22,8 @@ class WanAudioModel(WanModel):
def __init__(self, model_path, config, device):
self.config = config
self._load_adapter_ckpt()
self.run_device = self.config.get("run_device", "cuda")
self._load_adapter_ckpt()
super().__init__(model_path, config, device)
def _load_adapter_ckpt(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment