Spaces:
Running
on
Zero
Running
on
Zero
thuzhaowang
commited on
Commit
•
ff723b2
1
Parent(s):
7ad39d9
fix app
Browse files
app.py
CHANGED
@@ -16,6 +16,7 @@ from core.utils.math_utils import unnormalize_params
|
|
16 |
from huggingface_hub import hf_hub_download
|
17 |
|
18 |
# Setup PyTorch:
|
|
|
19 |
device = torch.device('cuda')
|
20 |
|
21 |
# Define the cache directory for model files
|
@@ -253,5 +254,5 @@ with gr.Blocks() as demo:
|
|
253 |
outputs=[output_model_obj],
|
254 |
)
|
255 |
|
256 |
-
demo.queue(
|
257 |
-
demo.launch(
|
|
|
16 |
from huggingface_hub import hf_hub_download
|
17 |
|
18 |
# Setup PyTorch:
|
19 |
+
torch.set_grad_enabled(False)
|
20 |
device = torch.device('cuda')
|
21 |
|
22 |
# Define the cache directory for model files
|
|
|
254 |
outputs=[output_model_obj],
|
255 |
)
|
256 |
|
257 |
+
demo.queue()
|
258 |
+
demo.launch()
|