Revert "Revert "move text encoder to GPU (#360)" (#361)" (#363)

This reverts commit 28ce927586.
2026-08-16 13:13:16 +02:00 · 2023-09-13 11:33:31 -07:00
parent 28ce927586
commit 53beede21d
2 changed files with 3 additions and 2 deletions
@@ -70,7 +70,8 @@ def sdxl_encode_adm_patched(self, **kwargs):


 def text_encoder_device_patched():
-    return torch.device("cpu")
+    # Fooocus's style system uses text encoder much more times than comfy so this makes things much faster.
+    return comfy.model_management.get_torch_device()


 def patch_all():