Unverified Commit fbb19ffc authored by sandy's avatar sandy Committed by GitHub
Browse files

[Bugfix] (#526)

parent 8e815fb8
...@@ -106,7 +106,7 @@ class HunyuanVideo15TransformerInfer(BaseTransformerInfer): ...@@ -106,7 +106,7 @@ class HunyuanVideo15TransformerInfer(BaseTransformerInfer):
self.seq_p_fp8_comm = self.config["parallel"].get("seq_p_fp8_comm", False) self.seq_p_fp8_comm = self.config["parallel"].get("seq_p_fp8_comm", False)
else: else:
self.seq_p_group = None self.seq_p_group = None
elf.seq_p_fp8_comm = False self.seq_p_fp8_comm = False
self.infer_func = self.infer_without_offload self.infer_func = self.infer_without_offload
if self.config.get("modulate_type", "triton") == "triton": if self.config.get("modulate_type", "triton") == "triton":
self.modulate_func = fuse_scale_shift_kernel self.modulate_func = fuse_scale_shift_kernel
......
...@@ -284,10 +284,9 @@ class DefaultRunner(BaseRunner): ...@@ -284,10 +284,9 @@ class DefaultRunner(BaseRunner):
if self.config.get("model_cls") == "wan2.2" and self.config["task"] in ["i2v", "s2v"]: if self.config.get("model_cls") == "wan2.2" and self.config["task"] in ["i2v", "s2v"]:
self.inputs["image_encoder_output"]["vae_encoder_out"] = None self.inputs["image_encoder_output"]["vae_encoder_out"] = None
if hasattr(self, "sr_version") and self.sr_version is not None is not None: if hasattr(self, "sr_version") and self.sr_version is not None:
self.lq_latents_shape = self.model.scheduler.latents.shape self.lq_latents_shape = self.model.scheduler.latents.shape
self.model_sr.set_scheduler(self.scheduler_sr) self.model_sr.set_scheduler(self.scheduler_sr)
self.config_sr["is_sr_running"] = True self.config_sr["is_sr_running"] = True
self.inputs_sr = self.run_input_encoder() self.inputs_sr = self.run_input_encoder()
self.config_sr["is_sr_running"] = False self.config_sr["is_sr_running"] = False
......
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