Update MagicQuill/comfy/sd.py

MagicQuill/comfy/sd.py

@@ -465,10 +465,17 @@ def load_clip(ckpt_paths, embedding_directory=None, clip_type=CLIPType.STABLE_DI
 def load_gligen(ckpt_path):
     data = comfy.utils.load_torch_file(ckpt_path, safe_load=True)
     model = gligen.load_gligen(data)
-    if model_management.should_use_fp16():
+
+    device = model_management.get_torch_device()
+    model = model.to(device)
+
+    if device.type == "cuda" and model_management.should_use_fp16():
         model = model.half()
-    return comfy.model_patcher.ModelPatcher(model, load_device=model_management.get_torch_device(), offload_device=model_management.unet_offload_device())
+    else:
+        model = model.float()
 
+    return comfy.model_patcher.ModelPatcher(model, load_device=device)
+    
 def load_checkpoint(config_path=None, ckpt_path=None, output_vae=True, output_clip=True, embedding_directory=None, state_dict=None, config=None):
     logging.warning("Warning: The load checkpoint with config function is deprecated and will eventually be removed, please use the other one.")
     model, clip, vae, _ = load_checkpoint_guess_config(ckpt_path, output_vae=output_vae, output_clip=output_clip, output_clipvision=False, embedding_directory=embedding_directory, output_model=True)