aatu18 commited on
Commit
c07f76d
·
verified ·
1 Parent(s): a89a6fc

retest with 30

Browse files
Files changed (1) hide show
  1. 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=10)
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')