Spaces:
Runtime error
Runtime error
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()
|