Spaces:
Running
on
Zero
Running
on
Zero
PseudoTerminal X
commited on
Commit
•
8fa8eb0
1
Parent(s):
6e6a0e5
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import spaces
|
2 |
from diffusers import DiffusionPipeline
|
3 |
|
4 |
-
pipe = DiffusionPipeline.from_pretrained(
|
5 |
pipe.to('cuda')
|
6 |
|
7 |
@spaces.GPU
|
|
|
1 |
import spaces
|
2 |
from diffusers import DiffusionPipeline
|
3 |
|
4 |
+
pipe = DiffusionPipeline.from_pretrained('ptx0/terminus-xl-velocity-v2')
|
5 |
pipe.to('cuda')
|
6 |
|
7 |
@spaces.GPU
|