298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
comfyanonymous 2023-08-28 15:08:45 -04:00
parent f92074b84f
commit 1300a1bb4c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/sd.py
View File

@ -80,7 +80,7 @@ class CLIP:
load_device = model_management.text_encoder_device() load_device = model_management.text_encoder_device()
offload_device = model_management.text_encoder_offload_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): if model_management.should_use_fp16(load_device, prioritize_performance=False):
params['dtype'] = torch.float16 params['dtype'] = torch.float16
else: else: