Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,12 +10,12 @@ from huggingface_hub import hf_hub_download
|
|
| 10 |
import spaces
|
| 11 |
from comfy import model_management
|
| 12 |
|
| 13 |
-
CHROMA_VERSION = "
|
| 14 |
|
| 15 |
# Download required models
|
| 16 |
t5_path = hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", filename="t5xxl_fp8_e4m3fn.safetensors", local_dir="models/text_encoders/")
|
| 17 |
vae_path = hf_hub_download(repo_id="lodestones/Chroma", filename="ae.safetensors", local_dir="models/vae")
|
| 18 |
-
unet_path = hf_hub_download(repo_id="lodestones/
|
| 19 |
|
| 20 |
# Example prompts with their parameters
|
| 21 |
EXAMPLES = [
|
|
|
|
| 10 |
import spaces
|
| 11 |
from comfy import model_management
|
| 12 |
|
| 13 |
+
CHROMA_VERSION = "chroma-unlocked-v48-detail-calibrated.safetensors"
|
| 14 |
|
| 15 |
# Download required models
|
| 16 |
t5_path = hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", filename="t5xxl_fp8_e4m3fn.safetensors", local_dir="models/text_encoders/")
|
| 17 |
vae_path = hf_hub_download(repo_id="lodestones/Chroma", filename="ae.safetensors", local_dir="models/vae")
|
| 18 |
+
unet_path = hf_hub_download(repo_id="lodestones/Chroma", filename=CHROMA_VERSION, local_dir="models/unet")
|
| 19 |
|
| 20 |
# Example prompts with their parameters
|
| 21 |
EXAMPLES = [
|