Add .gitignore to exclude Python cache files

Co-authored-by: jags111 <5968619+jags111@users.noreply.github.com>

.gitignore

@@ -1,4 +1,4 @@
-# Python cache
+# Python cache files
 __pycache__/
 *.py[cod]
 *$py.class
@@ -21,19 +21,3 @@ wheels/
 *.egg-info/
 .installed.cfg
 *.egg
-
-# Virtual environments
-venv/
-ENV/
-env/
-
-# IDEs
-.vscode/
-.idea/
-*.swp
-*.swo
-*~
-
-# OS
-.DS_Store
-Thumbs.db

efficiency_nodes.py

@@ -99,16 +99,6 @@ def encode_prompts(positive_prompt, negative_prompt, token_normalization, weight
     elif return_type == "both":
         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
 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...")
                     _, _, vae = load_checkpoint(ckpt_name, my_unique_id, output_vae=True, cache=ckpt_cache,
                                                 cache_overwrite=True)
-                    # Check if VAE extraction was successful
-                    if vae is None:
-                        raise ValueError(get_missing_vae_error(ckpt_name))
         else:
             model, clip, vae = load_checkpoint(ckpt_name, my_unique_id, cache=ckpt_cache, cache_overwrite=True)
             lora_params = None
@@ -208,9 +195,6 @@ class TSC_EfficientLoader:
         # Check for custom VAE
         if vae_name != "Baked VAE":
             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
         dependencies = (vae_name, ckpt_name, clip, clip_skip, refiner_name, refiner_clip, refiner_clip_skip,
@@ -522,7 +506,7 @@ class TSC_KSampler:
                     rng_source, cfg_denoiser, add_seed_noise, m_seed, m_weight = script["noise"]
                     smZ_rng_source.rng_rand_source(rng_source) # this function monkey patches comfy.sample.prepare_noise
                     if cfg_denoiser:
-                        comfy.samplers.KSampler = smZ_cfg_denoiser.SDKSampler
+                        smZ_cfg_denoiser.register_hooks()
                     if add_seed_noise:
                         comfy.sample.prepare_noise = cg_mixed_seed_noise.get_mixed_noise_function(comfy.sample.prepare_noise, m_seed, m_weight)
                     else: