Skip to content

Commit 8458ae2

Browse files
Revert "fix: run text encoders on MPS GPU instead of CPU for Apple Silicon (#…" (Comfy-Org#13070)
This reverts commit b941913.
1 parent fd0261d commit 8458ae2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy/model_management.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1003,7 +1003,7 @@ def text_encoder_offload_device():
10031003
def text_encoder_device():
10041004
if args.gpu_only:
10051005
return get_torch_device()
1006-
elif vram_state in (VRAMState.HIGH_VRAM, VRAMState.NORMAL_VRAM, VRAMState.SHARED) or comfy.memory_management.aimdo_enabled:
1006+
elif vram_state in (VRAMState.HIGH_VRAM, VRAMState.NORMAL_VRAM) or comfy.memory_management.aimdo_enabled:
10071007
if should_use_fp16(prioritize_performance=False):
10081008
return get_torch_device()
10091009
else:

0 commit comments

Comments
 (0)