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

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

View File

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

View File

@ -70,8 +70,7 @@ def sdxl_encode_adm_patched(self, **kwargs):
def text_encoder_device_patched():
# 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()
return torch.device("cpu")
def patch_all():