Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -398,7 +398,7 @@ def main():
|
|
| 398 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 399 |
|
| 400 |
pipe = YOSODiffusePipeline.from_pretrained(
|
| 401 |
-
'Stable-X/yoso-
|
| 402 |
torch_dtype=torch.float16, t_start=0).to(device)
|
| 403 |
try:
|
| 404 |
import xformers
|
|
|
|
| 398 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 399 |
|
| 400 |
pipe = YOSODiffusePipeline.from_pretrained(
|
| 401 |
+
'Stable-X/yoso-delight-v0-4', trust_remote_code=True, variant="fp16",
|
| 402 |
torch_dtype=torch.float16, t_start=0).to(device)
|
| 403 |
try:
|
| 404 |
import xformers
|