rework refiner for some potential new features (#642)

* sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync * sync
2026-08-16 13:13:16 +02:00 · 2023-10-11 03:07:43 -07:00
parent 777510dc9b
commit bbdf4bd120
10 changed files with 209 additions and 93 deletions
@@ -107,6 +107,10 @@ class ModelPatcher:
                for k in patch_list:
                    if hasattr(patch_list[k], "to"):
                        patch_list[k] = patch_list[k].to(device)
+        if "unet_wrapper_function" in self.model_options:
+            wrap_func = self.model_options["unet_wrapper_function"]
+            if hasattr(wrap_func, "to"):
+                self.model_options["unet_wrapper_function"] = wrap_func.to(device)

    def model_dtype(self):
        if hasattr(self.model, "get_dtype"):