"web/git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "086ac75228af3fea96be58ec45c2f6c17286e75d"
Commit 4a77fcd6 authored by comfyanonymous's avatar comfyanonymous
Browse files

Only shift text encoder to vram when CPU cores are under 8.

parent 3cd31d0e
...@@ -364,7 +364,8 @@ def text_encoder_device(): ...@@ -364,7 +364,8 @@ def text_encoder_device():
if args.gpu_only: if args.gpu_only:
return get_torch_device() return get_torch_device()
elif vram_state == VRAMState.HIGH_VRAM or vram_state == VRAMState.NORMAL_VRAM: elif vram_state == VRAMState.HIGH_VRAM or vram_state == VRAMState.NORMAL_VRAM:
if torch.get_num_threads() < 4: #leaving the text encoder on the CPU is faster than shifting it if the CPU is fast enough. #NOTE: on a Ryzen 5 7600X with 4080 it's faster to shift to GPU
if torch.get_num_threads() < 8: #leaving the text encoder on the CPU is faster than shifting it if the CPU is fast enough.
return get_torch_device() return get_torch_device()
else: else:
return torch.device("cpu") return torch.device("cpu")
......
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