x

scripts/sd_model_zoo.py

@@ -68,7 +68,7 @@ def load_image_encoder():
 def load_common_sd15_pipe(base_model=DEFAULT_BASE_MODEL, device="balanced", controlnet=None, ip_adapter=False, plus_model=True, torch_dtype=torch.float16, model_cpu_offload_seq=None, enable_sequential_cpu_offload=False, vae_slicing=False, pipeline_class=None, **kwargs):
     model_kwargs = dict(
         torch_dtype=torch_dtype, 
-        device_map=device,
+        # device_map=device,
         requires_safety_checker=False, 
         safety_checker=None,
     )