Update app.py

app.py

@@ -11,12 +11,12 @@ import spaces
 from comfy import model_management
 
 CHROMA_VERSION = "Chroma1-HD-flash-heun/Chroma1-HD-flash-heun-fp8_scaled_original_hybrid_large.safetensors"
-LORA_VERSION = "flash-heun-pruned/chroma-flash-heun_r01-fp32-pruned.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")
-lora_path = hf_hub_download(repo_id="silveroxides/Chroma-LoRAs", filename=LORA_VERSION, local_dir="models/loras")
+lora_path = hf_hub_download(repo_id="Alme995/DatasetTest_Loras_Chroma", filename=LORA_VERSION, local_dir="models/loras")
 
 # Example prompts with their parameters
 EXAMPLES = [