Skip to content

GitLab

"src/git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "cf6447eb2a68c7c8da3605d395fabb5ac186569f"
Commit 1300a1bb authored by comfyanonymous's avatar comfyanonymous
Browse files

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

parent f92074b8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/sd.py
View file @ 1300a1bb
...@@ -80,7 +80,7 @@ class CLIP: ...@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment