alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-21 21:22:13 -03:00
Code Issues Packages Projects Releases Wiki Activity

Improved readability

Browse Source
This commit is contained in:
Mijago
2023-07-26 22:07:16 +02:00
parent 6e67881e91
commit 264d553d9a
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
efficiency_nodes.py
View File

@@ -89,12 +89,20 @@ class TSC_EfficientLoader:
vae_cache, ckpt_cache, lora_cache = get_cache_numbers("Efficient Loader") vae_cache, ckpt_cache, lora_cache = get_cache_numbers("Efficient Loader")
if lora_name != "None" or lora_stack is not None: if lora_name != "None" or lora_stack is not None:
# Initialize an empty list to store LoRa parameters.
lora_params = [] lora_params = []
# Check if lora_name is not the string "None" and if so, add its parameters.
if lora_name != "None": if lora_name != "None":
lora_params.append((lora_name, lora_model_strength, lora_clip_strength)) lora_params.append((lora_name, lora_model_strength, lora_clip_strength))
if lora_stack is not None:
# If lora_stack is not None or an empty list, extend lora_params with its items.
if lora_stack:
lora_params.extend(lora_stack) lora_params.extend(lora_stack)
model, clip = load_lora(lora_params, ckpt_name, my_unique_id, cache=lora_cache, ckpt_cache=ckpt_cache, cache_overwrite=True)
# If there are any parameters in lora_params, load the LoRa with them.
if lora_params:
model, clip = load_lora(lora_params, ckpt_name, my_unique_id, cache=lora_cache, ckpt_cache=ckpt_cache, cache_overwrite=True)
if vae_name == "Baked VAE": if vae_name == "Baked VAE":
vae = get_bvae_by_ckpt_name(ckpt_name) vae = get_bvae_by_ckpt_name(ckpt_name)
else: else: