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

try fix lora vram (#357)

Browse Source 
* try fix lora vram

* try fix lora vram
This commit is contained in:
lllyasviel
2023-09-13 02:29:43 -07:00
committed by GitHub
parent eccf32b78c
commit a9b7219604
4 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/patch.py
+6
View File
@@ -3,6 +3,7 @@ import comfy.model_base
import comfy.ldm.modules.diffusionmodules.openaimodel
import comfy.samplers
import comfy.k_diffusion.external
import comfy.model_management
import modules.anisotropic as anisotropic
from comfy.k_diffusion import utils
@@ -68,6 +69,11 @@ def sdxl_encode_adm_patched(self, **kwargs):
return torch.cat((clip_pooled.to(flat.device), flat), dim=1)
def text_encoder_device_patched():
return torch.device("cpu")
def patch_all():
# comfy.model_management.text_encoder_device = text_encoder_device_patched
comfy.k_diffusion.external.DiscreteEpsDDPMDenoiser.forward = patched_discrete_eps_ddpm_denoiser_forward
comfy.model_base.SDXL.encode_adm = sdxl_encode_adm_patched