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

Revert "simplify codes"

Browse Source 
This reverts commit 2cc5aa7a3e.
This commit is contained in:
layerdiffusion
2024-08-14 20:39:00 -07:00
parent 2cc5aa7a3e
commit b31f81628f
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(patch_model_to)
self.real_model = self.model.forge_patch_model(device_to=patch_model_to)
except Exception as e:
self.model.forge_unpatch_model(self.model.offload_device)
self.model_unload()