Update app.py

app.py

@@ -10,10 +10,12 @@ from huggingface_hub import hf_hub_download
 import spaces
 from comfy import model_management
 
+CHROMA_VERSION = "chroma-unlocked-v32.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/")
 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-unlocked-v32.safetensors", local_dir="models/unet")
+unet_path = hf_hub_download(repo_id="lodestones/Chroma", filename=CHROMA_VERSION, local_dir="models/unet")
 
 # Utility functions
 def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
@@ -101,7 +103,7 @@ 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-unlocked-v31.safetensors", weight_dtype="fp8_e4m3fn"
+    unet_name=CHROMA_VERSION, weight_dtype="fp8_e4m3fn"
 )
 vaeloader_80 = vaeloader.load_vae(vae_name="ae.safetensors")