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 = "Chroma1-HD-flash-heun/Chroma1-HD-flash-heun-fp8_scaled_original_hybrid_large.safetensors"
+CHROMA_VERSION = "chroma2_kaleidoscope.safetensors"
 LORA_VERSION = "DatasetTest4.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-Misc-Models", filename=CHROMA_VERSION, local_dir="models/unet")
+unet_path = hf_hub_download(repo_id="silveroxides/Chroma2-Kaleidoscope", filename=CHROMA_VERSION, local_dir="models/unet")
 lora_path = hf_hub_download(repo_id="Alme995/DatasetTest_Loras_Chroma", filename=LORA_VERSION, local_dir="models/loras")
 
 # Example prompts with their parameters