From 1300a1bb4c87246b846c6209ab65ef0553004290 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Mon, 28 Aug 2023 15:08:45 -0400 Subject: [PATCH] Text encoder should initially load on the offload_device not the regular. --- comfy/sd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/sd.py b/comfy/sd.py index 1edc5fd2..a63a0d1d 100644 --- 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: