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

This reverts commit 28ce927586.
This commit is contained in:
lllyasviel 2023-09-13 11:33:31 -07:00 committed by GitHub
parent 28ce927586
commit 53beede21d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions

View File

@ -1 +1 @@
version = '2.0.3'
version = '2.0.4'

View File

@ -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():