From 986dd820dcae3b86a047b07a95f4d925fc625db0 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Mon, 13 Mar 2023 20:58:09 -0400
Subject: [PATCH] Use half() function on model when loading in fp16.

---
 comfy/sd.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy/sd.py b/comfy/sd.py
index 3f5ce24e..c7e0b073 100644
--- a/comfy/sd.py
+++ b/comfy/sd.py
@@ -854,4 +854,7 @@ def load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, e
     model = instantiate_from_config(model_config)
     model = load_model_weights(model, sd, verbose=False, load_state_dict_to=load_state_dict_to)
 
+    if fp16:
+        model = model.half()
+
     return (ModelPatcher(model), clip, vae)