soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

use better context manager to fix potential problems

Browse Source
This commit is contained in:
lllyasviel 2024-02-08 02:00:54 -08:00
parent 760f727eb9
commit 291ec743b6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules_forge/forge_loader.py
View File

@ -130,7 +130,7 @@ def load_checkpoint_guess_config(sd, output_vae=True, output_clip=True, output_c
return ForgeSD(model_patcher, clip, vae, clipvision)
@torch.inference_mode()
@torch.no_grad()
def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
a1111_config_filename = find_checkpoint_config(state_dict, checkpoint_info)
a1111_config = OmegaConf.load(a1111_config_filename)