Text encoder should initially load on the offload_device not the regular.

1300a1bb · comfyanonymous · f92074b8 · 1300a1bb
Commit 1300a1bb authored Aug 28, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy/sd.py comfy/sd.py +1 -1

No files found.
--- a/comfy/sd.py
+++ b/comfy/sd.py
@@ -80,7 +80,7 @@ class CLIP:

        load_device = model_management.text_encoder_device()
        offload_device = model_management.text_encoder_offload_device()
-        params['device'] = load_device
+        params['device'] = offload_device
        if model_management.should_use_fp16(load_device, prioritize_performance=False):
            params['dtype'] = torch.float16
        else: