Lower CPU thread check for running the text encoder on the CPU vs GPU.

This commit is contained in:
comfyanonymous 2023-07-30 17:18:24 -04:00
parent 2b13939044
commit 3cd31d0e24

View File

@ -364,7 +364,7 @@ def text_encoder_device():
if args.gpu_only:
return get_torch_device()
elif vram_state == VRAMState.HIGH_VRAM or vram_state == VRAMState.NORMAL_VRAM:
if torch.get_num_threads() < 8: #leaving the text encoder on the CPU is faster than shifting it if the CPU is fast enough.
if torch.get_num_threads() < 4: #leaving the text encoder on the CPU is faster than shifting it if the CPU is fast enough.
return get_torch_device()
else:
return torch.device("cpu")