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
|