Bug fix

2024-08-04 11:07:19 -06:00
parent f321de7bdb
commit 0ea27011d5
2 changed files with 10 additions and 2 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -468,8 +468,8 @@ class StableDiffusion:
                transformer_path,
                subfolder=subfolder,
                torch_dtype=dtype,
-                low_cpu_mem_usage=False,
-                device_map=None
+                # low_cpu_mem_usage=False,
+                # device_map=None
            )
            transformer.to(self.device_torch, dtype=dtype)
            flush()