cxllm-vendors/stable-diffusion-webui-forge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix ruff linter (#137)

Browse Source 
* Fix ruff linter

* Remove unused imports

* Remove unused imports
This commit is contained in:
Chenlei Hu
2024-02-09 01:35:20 +00:00
committed by GitHub
parent 66c22490c3
commit 6b3ad64388
18 changed files with 25 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules_forge/forge_loader.py
-2
View File
@@ -71,7 +71,6 @@ 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
@@ -99,7 +98,6 @@ 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.")