chroma

app.py

@@ -10,7 +10,15 @@ dtype = torch.bfloat16
 file_url = "https://huggingface.co/lodestones/Chroma/blob/main/chroma-unlocked-v31.safetensors"
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 
-flux_pipeline = FluxPipeline.from_single_file(file_url, torch_dtype=dtype, token=huggingface_token)
+# Load the pipeline with proper configuration
+flux_pipeline = FluxPipeline.from_single_file(
+    file_url,
+    torch_dtype=dtype,
+    token=huggingface_token,
+    use_safetensors=True,
+    local_files_only=False,
+    config_file="model_index.json"
+)
 flux_pipeline.to(device)
 
 @spaces.GPU()