Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/hytlass.git" did not exist on "d22dbec28b2dcc026b7c19a57ed71ce1ea9ed1b2"
Commit 2054eca3 authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

Update transformer_infer.py (#238)

parent 075ec4c0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lightx2v/models/networks/wan/infer/offload/transformer_infer.py
View file @ 2054eca3
...@@ -28,7 +28,7 @@ class WanOffloadTransformerInfer(WanTransformerInfer): ...@@ -28,7 +28,7 @@ class WanOffloadTransformerInfer(WanTransformerInfer):
else: else:
self.infer_func = self.infer_with_phases_lazy_offload self.infer_func = self.infer_with_phases_lazy_offload
elif offload_granularity == "model": elif offload_granularity == "model":
self.infer_func = self._infer_without_offload self.infer_func = self.infer_without_offload
if offload_granularity != "model": if offload_granularity != "model":
if not self.config.get("lazy_load", False): if not self.config.get("lazy_load", 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