CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

eps

Browse Source
This commit is contained in:
Haoming
2025-10-28 16:58:11 +08:00
parent 6069aaa95a
commit cfdfe3fb0a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+1 -1
View File
@@ -408,7 +408,7 @@ def apply_token_merging(sd_model, token_merging_ratio):
sd_model.forge_objects.unet = TomePatcher.patch(model=sd_model.forge_objects.unet, ratio=token_merging_ratio)
print(f"token_merging_ratio = {token_merging_ratio}")
if opts.scaling_factor > 1.0:
if opts.scaling_factor > 1.0 and sd_model.model_config.model_type.name == "EPS":
from backend.misc.eps import EpsilonScaling
sd_model.forge_objects.unet = EpsilonScaling.patch(model=sd_model.forge_objects.unet, scaling_factor=opts.scaling_factor)