lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-19 06:19:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

simplify codes

Browse Source
This commit is contained in:
layerdiffusion
2024-08-14 20:35:28 -07:00
parent e7567efd4b
commit 2cc5aa7a3e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/memory_management.py
View File

@@ -375,7 +375,7 @@ class LoadedModel:
self.model.model_patches_to(self.model.model_dtype())
try:
self.real_model = self.model.forge_patch_model(device_to=patch_model_to)
self.real_model = self.model.forge_patch_model(patch_model_to)
except Exception as e:
self.model.forge_unpatch_model(self.model.offload_device)
self.model_unload()