lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 03:01:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

unify cast name

Browse Source
This commit is contained in:
layerdiffusion
2024-07-30 08:42:51 -06:00
parent 9a48c9eff3
commit abd4d4d83d
2 changed files with 31 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ldm_patched/modules/model_management.py
View File

@@ -337,9 +337,9 @@ class LoadedModel:
real_async_memory = 0
mem_counter = 0
for m in self.real_model.modules():
if hasattr(m, "ldm_patched_cast_weights"):
m.prev_ldm_patched_cast_weights = m.ldm_patched_cast_weights
m.ldm_patched_cast_weights = True
if hasattr(m, "parameters_manual_cast"):
m.prev_parameters_manual_cast = m.parameters_manual_cast
m.parameters_manual_cast = True
module_mem = module_size(m)
if mem_counter + module_mem < async_kept_memory:
m.to(self.device)
@@ -366,9 +366,9 @@ class LoadedModel:
def model_unload(self, avoid_model_moving=False):
if self.model_accelerated:
for m in self.real_model.modules():
if hasattr(m, "prev_ldm_patched_cast_weights"):
m.ldm_patched_cast_weights = m.prev_ldm_patched_cast_weights
del m.prev_ldm_patched_cast_weights
if hasattr(m, "prev_parameters_manual_cast"):
m.parameters_manual_cast = m.prev_parameters_manual_cast
del m.prev_parameters_manual_cast
self.model_accelerated = False