Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

move text encoder to GPU (#360)

Browse Source
This commit is contained in:
lllyasviel
2023-09-13 03:43:15 -07:00
committed by GitHub
parent e0f97b6fef
commit 7700276b50
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fooocus_version.py
+1 -1
View File
@@ -1 +1 @@
version = '2.0.3'
version = '2.0.4'
modules/patch.py
+2 -1
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():