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 = "chroma-unlocked-v40.safetensors"
+CHROMA_VERSION = "staging_cfg_3/safetensors/2025-06-29_00-49-53.safetensors"
 
 # Download required models
-t5_path = hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", filename="t5xxl_fp8_e4m3fn.safetensors", local_dir="models/text_encoders/")
+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="lodestones/Chroma", filename=CHROMA_VERSION, local_dir="models/unet")
+unet_path = hf_hub_download(repo_id="silveroxides/chroma-debug-development-only-GGUF", filename=CHROMA_VERSION, local_dir="models/unet")
 
 # Example prompts with their parameters
 EXAMPLES = [
@@ -149,20 +149,20 @@ cliptextencode = CLIPTextEncode()
 unetloader = UNETLoader()
 vaeloader = VAELoader()
 cfgguider = NODE_CLASS_MAPPINGS["CFGGuider"]()
-basicscheduler = NODE_CLASS_MAPPINGS["BasicScheduler"]()
+betascheduler = NODE_CLASS_MAPPINGS["BetaSamplingScheduler"]()
 samplercustomadvanced = NODE_CLASS_MAPPINGS["SamplerCustomAdvanced"]()
 vaedecode = VAEDecode()
 saveimage = SaveImage()
 
 # Load models
 cliploader_78 = cliploader.load_clip(
-    clip_name="t5xxl_fp8_e4m3fn.safetensors", type="chroma", device="default"
+    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)
 )
 unetloader_76 = unetloader.load_unet(
-    unet_name=CHROMA_VERSION, weight_dtype="fp8_e4m3fn"
+    unet_name=CHROMA_VERSION, weight_dtype="default"
 )
 vaeloader_80 = vaeloader.load_vae(vae_name="ae.safetensors")
 
@@ -210,10 +210,10 @@ def generate_image(prompt, negative_prompt, width, height, steps, cfg, seed):
             negative=get_value_at_index(cliptextencode_75, 0),
         )
 
-        basicscheduler_84 = basicscheduler.get_sigmas(
-            scheduler="beta",
+        betascheduler_84 = betascheduler.get_sigmas(
             steps=steps,
-            denoise=1,
+            alpha=0.4,
+            beta=0.4,
             model=get_value_at_index(unetloader_76, 0),
         )
 
@@ -221,7 +221,7 @@ def generate_image(prompt, negative_prompt, width, height, steps, cfg, seed):
             noise=get_value_at_index(randomnoise_68, 0),
             guider=get_value_at_index(cfgguider_73, 0),
             sampler=get_value_at_index(ksamplerselect_72, 0),
-            sigmas=get_value_at_index(basicscheduler_84, 0),
+            sigmas=get_value_at_index(betascheduler_84, 0),
             latent_image=get_value_at_index(emptysd3latentimage_69, 0),
         )
 
@@ -293,10 +293,10 @@ Space Author: [GitHub](https://github.com/gokayfem) | [X.com](https://x.com/goka
                     label="Steps"
                 )
                 cfg = gr.Slider(
-                    minimum=1,
-                    maximum=20,
-                    value=4,
-                    step=0.5,
+                    minimum=0.1,
+                    maximum=4,
+                    value=1,
+                    step=0.1,
                     label="CFG Scale"
                 )
                 seed = gr.Number(