Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -110,7 +110,7 @@ class SimpleDiffusionPipeline(DiffusionPipeline):
|
|
| 110 |
#).to(device)
|
| 111 |
|
| 112 |
pipe_id = "AiArtLab/sdxs"
|
| 113 |
-
pipe =
|
| 114 |
pipe_id,
|
| 115 |
torch_dtype=dtype,
|
| 116 |
trust_remote_code=True
|
|
|
|
| 110 |
#).to(device)
|
| 111 |
|
| 112 |
pipe_id = "AiArtLab/sdxs"
|
| 113 |
+
pipe = DiffusionPipeline.from_pretrained(
|
| 114 |
pipe_id,
|
| 115 |
torch_dtype=dtype,
|
| 116 |
trust_remote_code=True
|