mirror of
https://github.com/jags111/efficiency-nodes-comfyui.git
synced 2026-05-07 09:06:44 -03:00
Compare commits
3 Commits
copilot/fi
...
copilot/fi
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4ee7e8bfd4 | ||
|
|
292b444099 | ||
|
|
ffa6fde8f3 |
45
.gitignore
vendored
45
.gitignore
vendored
@@ -1,39 +1,10 @@
|
|||||||
# Python cache
|
# Python cache files
|
||||||
__pycache__/
|
__pycache__/
|
||||||
*.py[cod]
|
*.pyc
|
||||||
*$py.class
|
*.pyo
|
||||||
|
*.pyd
|
||||||
|
|
||||||
|
# Compiled files
|
||||||
*.so
|
*.so
|
||||||
|
*.dll
|
||||||
# Distribution / packaging
|
*.dylib
|
||||||
.Python
|
|
||||||
build/
|
|
||||||
develop-eggs/
|
|
||||||
dist/
|
|
||||||
downloads/
|
|
||||||
eggs/
|
|
||||||
.eggs/
|
|
||||||
lib/
|
|
||||||
lib64/
|
|
||||||
parts/
|
|
||||||
sdist/
|
|
||||||
var/
|
|
||||||
wheels/
|
|
||||||
*.egg-info/
|
|
||||||
.installed.cfg
|
|
||||||
*.egg
|
|
||||||
|
|
||||||
# Virtual environments
|
|
||||||
venv/
|
|
||||||
ENV/
|
|
||||||
env/
|
|
||||||
|
|
||||||
# IDEs
|
|
||||||
.vscode/
|
|
||||||
.idea/
|
|
||||||
*.swp
|
|
||||||
*.swo
|
|
||||||
*~
|
|
||||||
|
|
||||||
# OS
|
|
||||||
.DS_Store
|
|
||||||
Thumbs.db
|
|
||||||
|
|||||||
@@ -99,16 +99,6 @@ def encode_prompts(positive_prompt, negative_prompt, token_normalization, weight
|
|||||||
elif return_type == "both":
|
elif return_type == "both":
|
||||||
return positive_encoded, negative_encoded, clip, refiner_positive_encoded, refiner_negative_encoded, refiner_clip
|
return positive_encoded, negative_encoded, clip, refiner_positive_encoded, refiner_negative_encoded, refiner_clip
|
||||||
|
|
||||||
########################################################################################################################
|
|
||||||
# Helper function for VAE error message
|
|
||||||
def get_missing_vae_error(ckpt_name):
|
|
||||||
"""Generate error message for checkpoints without embedded VAE"""
|
|
||||||
return (
|
|
||||||
f"Checkpoint '{ckpt_name}' does not contain an embedded VAE. "
|
|
||||||
f"This checkpoint (likely an AIO model) requires an external VAE. "
|
|
||||||
f"Please select a VAE file instead of 'Baked VAE'."
|
|
||||||
)
|
|
||||||
|
|
||||||
########################################################################################################################
|
########################################################################################################################
|
||||||
# TSC Efficient Loader
|
# TSC Efficient Loader
|
||||||
class TSC_EfficientLoader:
|
class TSC_EfficientLoader:
|
||||||
@@ -176,9 +166,6 @@ class TSC_EfficientLoader:
|
|||||||
f"{warning('Efficiency Nodes:')} Baked VAE not found in cache, loading checkpoint to extract VAE...")
|
f"{warning('Efficiency Nodes:')} Baked VAE not found in cache, loading checkpoint to extract VAE...")
|
||||||
_, _, vae = load_checkpoint(ckpt_name, my_unique_id, output_vae=True, cache=ckpt_cache,
|
_, _, vae = load_checkpoint(ckpt_name, my_unique_id, output_vae=True, cache=ckpt_cache,
|
||||||
cache_overwrite=True)
|
cache_overwrite=True)
|
||||||
# Check if VAE extraction was successful
|
|
||||||
if vae is None:
|
|
||||||
raise ValueError(get_missing_vae_error(ckpt_name))
|
|
||||||
else:
|
else:
|
||||||
model, clip, vae = load_checkpoint(ckpt_name, my_unique_id, cache=ckpt_cache, cache_overwrite=True)
|
model, clip, vae = load_checkpoint(ckpt_name, my_unique_id, cache=ckpt_cache, cache_overwrite=True)
|
||||||
lora_params = None
|
lora_params = None
|
||||||
@@ -208,9 +195,6 @@ class TSC_EfficientLoader:
|
|||||||
# Check for custom VAE
|
# Check for custom VAE
|
||||||
if vae_name != "Baked VAE":
|
if vae_name != "Baked VAE":
|
||||||
vae = load_vae(vae_name, my_unique_id, cache=vae_cache, cache_overwrite=True)
|
vae = load_vae(vae_name, my_unique_id, cache=vae_cache, cache_overwrite=True)
|
||||||
elif vae is None:
|
|
||||||
# If "Baked VAE" was selected but checkpoint has no embedded VAE
|
|
||||||
raise ValueError(get_missing_vae_error(ckpt_name))
|
|
||||||
|
|
||||||
# Data for XY Plot
|
# Data for XY Plot
|
||||||
dependencies = (vae_name, ckpt_name, clip, clip_skip, refiner_name, refiner_clip, refiner_clip_skip,
|
dependencies = (vae_name, ckpt_name, clip, clip_skip, refiner_name, refiner_clip, refiner_clip_skip,
|
||||||
@@ -342,6 +326,10 @@ class TSC_LoRA_Stacker:
|
|||||||
FUNCTION = "lora_stacker"
|
FUNCTION = "lora_stacker"
|
||||||
CATEGORY = "Efficiency Nodes/Stackers"
|
CATEGORY = "Efficiency Nodes/Stackers"
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def IS_CHANGED(cls, **kwargs):
|
||||||
|
return float("nan")
|
||||||
|
|
||||||
def lora_stacker(self, input_mode, lora_count, lora_stack=None, **kwargs):
|
def lora_stacker(self, input_mode, lora_count, lora_stack=None, **kwargs):
|
||||||
|
|
||||||
# Extract values from kwargs
|
# Extract values from kwargs
|
||||||
|
|||||||
Reference in New Issue
Block a user