Skip to content

GitLab

Commit 7e455adc authored by gameltb's avatar gameltb
Browse files

fix unet_wrapper_function name in ModelPatcher

parent 1ffa8858
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
comfy/model_patcher.py
View file @ 7e455adc
...@@ -107,10 +107,10 @@ class ModelPatcher: ...@@ -107,10 +107,10 @@ class ModelPatcher:
for k in patch_list: for k in patch_list:
if hasattr(patch_list[k], "to"): if hasattr(patch_list[k], "to"):
patch_list[k] = patch_list[k].to(device) patch_list[k] = patch_list[k].to(device)
if "unet_wrapper_function" in self.model_options: if "model_function_wrapper" in self.model_options:
wrap_func = self.model_options["unet_wrapper_function"] wrap_func = self.model_options["model_function_wrapper"]
if hasattr(wrap_func, "to"): if hasattr(wrap_func, "to"):
self.model_options["unet_wrapper_function"] = wrap_func.to(device) self.model_options["model_function_wrapper"] = wrap_func.to(device)
def model_dtype(self): def model_dtype(self):
if hasattr(self.model, "get_dtype"): if hasattr(self.model, "get_dtype"):
......
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