Update app.py

app.py

@@ -10,12 +10,12 @@ from huggingface_hub import hf_hub_download
 import spaces
 from comfy import model_management
 
-CHROMA_VERSION = "staging_cfg_3/safetensors/2025-06-29_00-49-53.safetensors"
+CHROMA_VERSION = "Chroma1-HD-flash-heun/Chroma1-HD-flash-heun-fp8_scaled_original_hybrid_large.safetensors"
 
 # Download required models
 t5_path = hf_hub_download(repo_id="silveroxides/flan-t5-xxl-encoder-only", filename="flan-t5-xxl_float8_e4m3fn_scaled_stochastic.safetensors", local_dir="models/text_encoders/")
 vae_path = hf_hub_download(repo_id="lodestones/Chroma", filename="ae.safetensors", local_dir="models/vae")
-unet_path = hf_hub_download(repo_id="silveroxides/chroma-debug-development-only-GGUF", filename=CHROMA_VERSION, local_dir="models/unet")
+unet_path = hf_hub_download(repo_id="silveroxides/Chroma-Misc-Models", filename=CHROMA_VERSION, local_dir="models/unet")
 
 # Example prompts with their parameters
 EXAMPLES = [
@@ -138,7 +138,7 @@ from nodes import (
     VAELoader,
     SaveImage,
 )
-
+from custom_nodes.
 # Initialize all model loaders outside the function
 randomnoise = NODE_CLASS_MAPPINGS["RandomNoise"]()
 emptysd3latentimage = NODE_CLASS_MAPPINGS["EmptySD3LatentImage"]()
@@ -146,10 +146,10 @@ ksamplerselect = NODE_CLASS_MAPPINGS["KSamplerSelect"]()
 cliploader = CLIPLoader()
 t5tokenizeroptions = NODE_CLASS_MAPPINGS["T5TokenizerOptions"]()
 cliptextencode = CLIPTextEncode()
-unetloader = UNETLoader()
+unetloader = NODE_CLASS_MAPPINGS["ScaledFP8HybridUNetLoader"]()
 vaeloader = VAELoader()
-modelsamplingflux = NODE_CLASS_MAPPINGS["ModelSamplingFlux"]()
-cfgguider = NODE_CLASS_MAPPINGS["CFGGuider"]()
+modelsamplingauraflow = NODE_CLASS_MAPPINGS["ModelSamplingAuraflow"]()
+basicguider = NODE_CLASS_MAPPINGS["BasicGuider"]()
 betascheduler = NODE_CLASS_MAPPINGS["BetaSamplingScheduler"]()
 samplercustomadvanced = NODE_CLASS_MAPPINGS["SamplerCustomAdvanced"]()
 vaedecode = VAEDecode()
@@ -160,10 +160,10 @@ cliploader_78 = cliploader.load_clip(
     clip_name="flan-t5-xxl_float8_e4m3fn_scaled_stochastic.safetensors", type="chroma", device="default"
 )
 t5tokenizeroptions_82 = t5tokenizeroptions.set_options(
-    min_padding=1, min_length=0, clip=get_value_at_index(cliploader_78, 0)
+    min_padding=0, min_length=0, clip=get_value_at_index(cliploader_78, 0)
 )
 unetloader_76 = unetloader.load_unet(
-    unet_name=CHROMA_VERSION, weight_dtype="default"
+    model_name=CHROMA_VERSION, chroma_hybrid_large=True, radiance_hybrid_large=False, chroma_hybrid_small=False, radiance_hybrid_small=False, wan=False, qwen=False
 )
 vaeloader_80 = vaeloader.load_vae(vae_name="ae.safetensors")
 
@@ -192,43 +192,34 @@ def generate_image(prompt, negative_prompt, width, height, steps, cfg, seed):
         emptysd3latentimage_69 = emptysd3latentimage.generate(
             width=width, height=height, batch_size=1
         )
-        ksamplerselect_72 = ksamplerselect.get_sampler(sampler_name="res_multistep")
+        ksamplerselect_72 = ksamplerselect.get_sampler(sampler_name="dpmpp_sde")
 
         cliptextencode_74 = cliptextencode.encode(
             text=prompt,
             clip=get_value_at_index(t5tokenizeroptions_82, 0),
         )
 
-        cliptextencode_75 = cliptextencode.encode(
-            text=negative_prompt,
-            clip=get_value_at_index(t5tokenizeroptions_82, 0),
-        )
 
-        modelsamplingflux_85 = modelsamplingflux.patch(
-            max_shift=1.0,
-            base_shift=0.0,
-            width=width,
-            height=height,
+        modelsamplingauraflow_85 = modelsamplingauraflow.patch(
+            shift=1.0,
             model=get_value_at_index(unetloader_76, 0),
         )
 
-        cfgguider_73 = cfgguider.get_guider(
-            cfg=cfg,
+        basicguider_73 = basicguider.get_guider(
             model=get_value_at_index(unetloader_76, 0),
             positive=get_value_at_index(cliptextencode_74, 0),
-            negative=get_value_at_index(cliptextencode_75, 0),
         )
 
         betascheduler_84 = betascheduler.get_sigmas(
             steps=steps,
-            alpha=0.55,
-            beta=0.4,
+            alpha=0.45,
+            beta=0.45,
             model=get_value_at_index(unetloader_76, 0),
         )
 
         samplercustomadvanced_67 = samplercustomadvanced.sample(
             noise=get_value_at_index(randomnoise_68, 0),
-            guider=get_value_at_index(cfgguider_73, 0),
+            guider=get_value_at_index(basicguider_73, 0),
             sampler=get_value_at_index(ksamplerselect_72, 0),
             sigmas=get_value_at_index(betascheduler_84, 0),
             latent_image=get_value_at_index(emptysd3latentimage_69, 0),