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

fix many problems

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 09:28:38 -08:00
parent dcdc80810e
commit 7b14cb3f9f
7 changed files with 208 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
modules_forge/patch_basic.py
View File

@@ -1,4 +1,5 @@
import torch
import time
import ldm_patched.modules.samplers
from ldm_patched.modules.controlnet import ControlBase
@@ -183,7 +184,21 @@ def patched_calc_cond_uncond_batch(model, cond, uncond, x_in, timestep, model_op
return out_cond, out_uncond
def patched_load_models_gpu(*args, **kwargs):
execution_start_time = time.perf_counter()
y = model_management.load_models_gpu_origin(*args, **kwargs)
moving_time = time.perf_counter() - execution_start_time
if moving_time > 0.1:
print(f'Moving model(s) has taken {moving_time:.2f} seconds')
return y
def patch_all_basics():
if not hasattr(model_management, 'load_models_gpu_origin'):
model_management.load_models_gpu_origin = model_management.load_models_gpu
model_management.load_models_gpu = patched_load_models_gpu
ControlBase.control_merge = patched_control_merge
ldm_patched.modules.samplers.calc_cond_uncond_batch = patched_calc_cond_uncond_batch
return