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

less verbose

Browse Source
This commit is contained in:
lllyasviel
2024-02-23 22:31:16 -08:00
parent 2ecb869f31
commit 79bdb78619
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ldm_patched/modules/model_management.py
View File

@@ -328,7 +328,6 @@ class LoadedModel:
if not disable_async_load: if not disable_async_load:
print("[Memory Management] Requested Async Preserved Memory (MB) = ", async_kept_memory / (1024 * 1024)) print("[Memory Management] Requested Async Preserved Memory (MB) = ", async_kept_memory / (1024 * 1024))
real_async_memory = 0 real_async_memory = 0
real_kept_memory = 0
mem_counter = 0 mem_counter = 0
for m in self.real_model.modules(): for m in self.real_model.modules():
if hasattr(m, "ldm_patched_cast_weights"): if hasattr(m, "ldm_patched_cast_weights"):
@@ -338,7 +337,6 @@ class LoadedModel:
if mem_counter + module_mem < async_kept_memory: if mem_counter + module_mem < async_kept_memory:
m.to(self.device) m.to(self.device)
mem_counter += module_mem mem_counter += module_mem
real_kept_memory += module_mem
else: else:
real_async_memory += module_mem real_async_memory += module_mem
m.to(self.model.offload_device) m.to(self.model.offload_device)
@@ -349,7 +347,7 @@ class LoadedModel:
mem_counter += module_size(m) mem_counter += module_size(m)
print("[Memory Management] Async Loader Disabled for ", m) print("[Memory Management] Async Loader Disabled for ", m)
print("[Async Memory Management] Parameters Loaded to Async Stream (MB) = ", real_async_memory / (1024 * 1024)) print("[Async Memory Management] Parameters Loaded to Async Stream (MB) = ", real_async_memory / (1024 * 1024))
print("[Async Memory Management] Parameters Loaded to GPU (MB) = ", real_kept_memory / (1024 * 1024)) print("[Async Memory Management] Parameters Loaded to GPU (MB) = ", mem_counter / (1024 * 1024))
self.model_accelerated = True self.model_accelerated = True