move expansion to managed device (#364)

* move expansion to managed device

* move expansion to managed device

* move expansion to managed device

* move expansion to managed device

* move expansion to managed device

* move expansion to managed device
This commit is contained in:
lllyasviel
2023-09-13 12:48:27 -07:00
committed by GitHub
parent 53beede21d
commit e32f04da34
4 changed files with 17 additions and 3 deletions
+2
View File
@@ -76,5 +76,7 @@ def text_encoder_device_patched():
def patch_all():
comfy.model_management.text_encoder_device = text_encoder_device_patched
print(f'Fooocus Text Processing Pipelines are retargeted to {str(comfy.model_management.text_encoder_device())}')
comfy.k_diffusion.external.DiscreteEpsDDPMDenoiser.forward = patched_discrete_eps_ddpm_denoiser_forward
comfy.model_base.SDXL.encode_adm = sdxl_encode_adm_patched