From 291ec743b603fdcd9c58e60dc5ed3d866c53bc4c Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 8 Feb 2024 02:00:54 -0800 Subject: [PATCH] use better context manager to fix potential problems --- modules_forge/forge_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index caf52735..d662f8dd 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -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)