Spaces:
Sleeping
Sleeping
retest with 30
Browse files- gdc_pipeline.py +1 -1
gdc_pipeline.py
CHANGED
|
@@ -128,7 +128,7 @@ def construct_and_execute_api_call(
|
|
| 128 |
|
| 129 |
|
| 130 |
# generate llama model response
|
| 131 |
-
@spaces.GPU(duration=
|
| 132 |
def generate_response(modified_query, model, tok):
|
| 133 |
set_seed(1042)
|
| 134 |
# model = model.to('cuda' if torch.cuda.is_available() else 'cpu')
|
|
|
|
| 128 |
|
| 129 |
|
| 130 |
# generate llama model response
|
| 131 |
+
@spaces.GPU(duration=30)
|
| 132 |
def generate_response(modified_query, model, tok):
|
| 133 |
set_seed(1042)
|
| 134 |
# model = model.to('cuda' if torch.cuda.is_available() else 'cpu')
|