Chroma-Extra

Running on Zero

gokaygokay commited on May 23

Commit

859912a

verified ·

1 Parent(s): babf748

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from diffusers import FluxPipeline
 # Initialize model outside the function
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16
-file_url = "https://huggingface.co/lodestones/Chroma/resolve/main/chroma-unlocked-v31.safetensors"
 flux_pipeline = FluxPipeline.from_single_file(file_url, torch_dtype=dtype)
 flux_pipeline.to(device)

 # Initialize model outside the function
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16
+file_url = "https://huggingface.co/lodestones/Chroma/blob/main/chroma-unlocked-v31.safetensors"
 flux_pipeline = FluxPipeline.from_single_file(file_url, torch_dtype=dtype)
 flux_pipeline.to(device)