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

compile

Browse Source
This commit is contained in:
Haoming 2026-02-19 22:42:45 +08:00
parent aeb5600e17
commit 863319a826
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
extensions-builtin/sd_forge_compile/scripts/compile.py
View File

@ -75,8 +75,10 @@ class TorchCompileForForge(scripts.Script):
else:
config = dict(backend=backend, dynamic=True, fullgraph=True)
_cache: str = str(config) + p.sd_model.current_lora_hash
if (_config := getattr(kmodel, "_compile_config", None)) is not None:
if _config != config:
if _config != _cache:
self.restore(kmodel)
else:
return
@ -90,4 +92,4 @@ class TorchCompileForForge(scripts.Script):
torch.compile(model, **config),
)
kmodel._compile_config = config
kmodel._compile_config = _cache