Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
3b62d0f
1
Parent(s):
824bcc3
chroma
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from transformers import T5EncoderModel, BitsAndBytesConfig as BitsAndBytesConfi
|
|
8 |
# Initialize model outside the function
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
10 |
dtype = torch.bfloat16
|
11 |
-
file_url = "https://huggingface.co/lodestones/Chroma/
|
12 |
huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
|
13 |
single_file_base_model = "camenduru/FLUX.1-dev-diffusers"
|
14 |
|
|
|
8 |
# Initialize model outside the function
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
10 |
dtype = torch.bfloat16
|
11 |
+
file_url = "https://huggingface.co/lodestones/Chroma/blob/main/chroma-unlocked-v31.safetensors"
|
12 |
huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
|
13 |
single_file_base_model = "camenduru/FLUX.1-dev-diffusers"
|
14 |
|