Spaces:
Running
on
A10G
Running
on
A10G
Ava Pun
commited on
Commit
·
7825517
1
Parent(s):
cc9cd8f
increase duration
Browse files
app.py
CHANGED
|
@@ -10,7 +10,6 @@ from urllib.request import urlretrieve
|
|
| 10 |
|
| 11 |
import gradio as gr
|
| 12 |
import spaces
|
| 13 |
-
import torch.multiprocessing as mp
|
| 14 |
import transformers
|
| 15 |
from brickgpt.models import BrickGPT, BrickGPTConfig
|
| 16 |
|
|
@@ -104,7 +103,7 @@ class BrickGenerator:
|
|
| 104 |
self.model_cfg = model_cfg
|
| 105 |
self.model = None
|
| 106 |
|
| 107 |
-
@spaces.GPU
|
| 108 |
def generate_bricks(
|
| 109 |
self,
|
| 110 |
prompt: str,
|
|
|
|
| 10 |
|
| 11 |
import gradio as gr
|
| 12 |
import spaces
|
|
|
|
| 13 |
import transformers
|
| 14 |
from brickgpt.models import BrickGPT, BrickGPTConfig
|
| 15 |
|
|
|
|
| 103 |
self.model_cfg = model_cfg
|
| 104 |
self.model = None
|
| 105 |
|
| 106 |
+
@spaces.GPU(duration=600)
|
| 107 |
def generate_bricks(
|
| 108 |
self,
|
| 109 |
prompt: str,
|