Spaces:
Sleeping
Sleeping
Yinhong Liu
commited on
Commit
·
347bb62
1
Parent(s):
64f514c
sana dtype
Browse files
app.py
CHANGED
|
@@ -18,6 +18,7 @@ MODEL_OPTIONS = {
|
|
| 18 |
"SiD-Flow-Sana-0.6B-512-res": "YGu1998/SiD-Flow-Sana-0.6B-512-res",
|
| 19 |
"SiDA-Flow-Sana-0.6B-512-res": "YGu1998/SiDA-Flow-Sana-0.6B-512-res",
|
| 20 |
"SiD-Flow-Sana-1.6B-512-res": "YGu1998/SiD-Flow-Sana-1.6B-512-res",
|
|
|
|
| 21 |
"SiD-Flow-Sana-Sprint-0.6B-1024-res": "YGu1998/SiD-Flow-Sana-Sprint-0.6B-1024-res",
|
| 22 |
"SiDA-Flow-Sana-Sprint-0.6B-1024-res": "YGu1998/SiDA-Flow-Sana-Sprint-0.6B-1024-res",
|
| 23 |
"SiD-Flow-Sana-Sprint-1.6B-1024-res": "YGu1998/SiD-Flow-Sana-Sprint-1.6B-1024-res",
|
|
@@ -31,7 +32,7 @@ def load_model(model_choice):
|
|
| 31 |
model_repo_id = MODEL_OPTIONS[model_choice]
|
| 32 |
time_scale = 1000.0
|
| 33 |
if "Sana" in model_choice:
|
| 34 |
-
pipe = SiDSanaPipeline.from_pretrained(model_repo_id, torch_dtype=
|
| 35 |
if "Sprint" in model_choice:
|
| 36 |
time_scale = 1.0
|
| 37 |
elif "SD3" in model_choice:
|
|
|
|
| 18 |
"SiD-Flow-Sana-0.6B-512-res": "YGu1998/SiD-Flow-Sana-0.6B-512-res",
|
| 19 |
"SiDA-Flow-Sana-0.6B-512-res": "YGu1998/SiDA-Flow-Sana-0.6B-512-res",
|
| 20 |
"SiD-Flow-Sana-1.6B-512-res": "YGu1998/SiD-Flow-Sana-1.6B-512-res",
|
| 21 |
+
"SiDA-Flow-Sana-1.6B-512-res": "YGu1998/SiDA-Flow-Sana-1.6B-512-res",
|
| 22 |
"SiD-Flow-Sana-Sprint-0.6B-1024-res": "YGu1998/SiD-Flow-Sana-Sprint-0.6B-1024-res",
|
| 23 |
"SiDA-Flow-Sana-Sprint-0.6B-1024-res": "YGu1998/SiDA-Flow-Sana-Sprint-0.6B-1024-res",
|
| 24 |
"SiD-Flow-Sana-Sprint-1.6B-1024-res": "YGu1998/SiD-Flow-Sana-Sprint-1.6B-1024-res",
|
|
|
|
| 32 |
model_repo_id = MODEL_OPTIONS[model_choice]
|
| 33 |
time_scale = 1000.0
|
| 34 |
if "Sana" in model_choice:
|
| 35 |
+
pipe = SiDSanaPipeline.from_pretrained(model_repo_id, torch_dtype=torch.bfloat16)
|
| 36 |
if "Sprint" in model_choice:
|
| 37 |
time_scale = 1.0
|
| 38 |
elif "SD3" in model_choice:
|