ejschwartz commited on
Commit
9002ae5
·
1 Parent(s): 7995ffa

Try longer timeout

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def decomp_create_prompt(input_data: str) -> str:
18
  pipe = pipeline(model="ejschwartz/decaf-v1-22b-4bit", return_full_text=False)
19
  pipe.model.to("cuda")
20
 
21
- @spaces.GPU
22
  def generate(text):
23
  print(f"Generating text... {text}")
24
  prompt = decomp_create_prompt(text)
 
18
  pipe = pipeline(model="ejschwartz/decaf-v1-22b-4bit", return_full_text=False)
19
  pipe.model.to("cuda")
20
 
21
+ @spaces.GPU(duration=180)
22
  def generate(text):
23
  print(f"Generating text... {text}")
24
  prompt = decomp_create_prompt(text)