Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
9b4a79f
1
Parent(s):
df7b4b1
new image models
Browse files
app.py
CHANGED
|
@@ -7,12 +7,16 @@ from diffusers import DiffusionPipeline
|
|
| 7 |
import torch
|
| 8 |
|
| 9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 10 |
-
model_repo_id = "
|
| 11 |
|
| 12 |
MODEL_CHOICES = [
|
| 13 |
-
"
|
| 14 |
-
"
|
| 15 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
]
|
| 17 |
|
| 18 |
if torch.cuda.is_available():
|
|
|
|
| 7 |
import torch
|
| 8 |
|
| 9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 10 |
+
model_repo_id = "lodestones/Chroma1-HD" # Replace to the model you would like to use
|
| 11 |
|
| 12 |
MODEL_CHOICES = [
|
| 13 |
+
"lodestones/Chroma1-HD",
|
| 14 |
+
"black-forest-labs/FLUX.1-dev",
|
| 15 |
+
"Qwen/Qwen-Image",
|
| 16 |
+
"tencent/HunyuanImage-3.0",
|
| 17 |
+
"stabilityai/stable-diffusion-3.5-medium",
|
| 18 |
+
"stabilityai/stable-diffusion-2-1",
|
| 19 |
+
"SG161222/Realistic_Vision_V5.1_noVAE"
|
| 20 |
]
|
| 21 |
|
| 22 |
if torch.cuda.is_available():
|