"vscode:/vscode.git/clone" did not exist on "87842db48e71c48bbad90d581f12924496ee0798"
Commit 3cd31d0e authored by comfyanonymous's avatar comfyanonymous
Browse files

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

parent 2b139390
...@@ -364,7 +364,7 @@ def text_encoder_device(): ...@@ -364,7 +364,7 @@ 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() < 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() 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