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

Revert "Fix ruff linter (#137)" (#143)

Browse Source 
This reverts commit 6b3ad64388.
This commit is contained in:
Chenlei Hu
2024-02-09 02:24:04 +00:00
committed by GitHub
parent b49742354d
commit 388ca351f4
18 changed files with 39 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules_forge/forge_loader.py
View File

@@ -71,6 +71,7 @@ def no_clip():
def load_checkpoint_guess_config(sd, output_vae=True, output_clip=True, output_clipvision=False, embedding_directory=None, output_model=True):
sd_keys = sd.keys()
clip = None
clipvision = None
vae = None
@@ -98,6 +99,7 @@ def load_checkpoint_guess_config(sd, output_vae=True, output_clip=True, output_c
if output_model:
inital_load_device = model_management.unet_inital_load_device(parameters, unet_dtype)
offload_device = model_management.unet_offload_device()
model = model_config.get_model(sd, "model.diffusion_model.", device=inital_load_device)
model.load_model_weights(sd, "model.diffusion_model.")