Spaces:
Running
on
T4
Running
on
T4
Commit
•
5c51951
1
Parent(s):
bc26ef9
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ notice = "<p><center>Acknowledgment: This demo is powered by GPU resources suppo
|
|
| 29 |
|
| 30 |
# setup
|
| 31 |
@spaces.GPU
|
| 32 |
-
def setup(
|
| 33 |
device = ("cuda" if torch.cuda.is_available() else "cpu")
|
| 34 |
img_pipe = DDPMPipeline.from_pretrained("yuntian-deng/latex2im_ss_finetunegptneo")
|
| 35 |
img_pipe.to(device)
|
|
|
|
| 29 |
|
| 30 |
# setup
|
| 31 |
@spaces.GPU
|
| 32 |
+
def setup():
|
| 33 |
device = ("cuda" if torch.cuda.is_available() else "cpu")
|
| 34 |
img_pipe = DDPMPipeline.from_pretrained("yuntian-deng/latex2im_ss_finetunegptneo")
|
| 35 |
img_pipe.to(device)
|