Skip to content

GitLab

Unverified Commit e251e4dc authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

support old r2v model (#337)

parent 4929cd02
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lightx2v/models/runners/wan/wan_audio_runner.py
View file @ e251e4dc
...@@ -787,7 +787,7 @@ class WanAudioRunner(WanRunner): # type:ignore ...@@ -787,7 +787,7 @@ class WanAudioRunner(WanRunner): # type:ignore
return base_model return base_model
def load_audio_encoder(self): def load_audio_encoder(self):
audio_encoder_path = os.path.join(self.config["model_path"], "TencentGameMate-chinese-hubert-large") audio_encoder_path = self.config.get("audio_encoder_path", os.path.join(self.config["model_path"], "TencentGameMate-chinese-hubert-large"))
audio_encoder_offload = self.config.get("audio_encoder_cpu_offload", self.config.get("cpu_offload", False)) audio_encoder_offload = self.config.get("audio_encoder_cpu_offload", self.config.get("cpu_offload", False))
model = SekoAudioEncoderModel(audio_encoder_path, self.config["audio_sr"], audio_encoder_offload) model = SekoAudioEncoderModel(audio_encoder_path, self.config["audio_sr"], audio_encoder_offload)
return model return model
......
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