karths commited on
Commit
668a83f
·
verified ·
1 Parent(s): 073094f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ if llama_tokenizer.pad_token is None:
102
  llama_tokenizer.pad_token = llama_tokenizer.eos_token
103
 
104
 
105
- @spaces.GPU(duration=90)
106
  def llama_generate(
107
  message: str,
108
  max_new_tokens: int = LLAMA_DEFAULT_MAX_NEW_TOKENS,
 
102
  llama_tokenizer.pad_token = llama_tokenizer.eos_token
103
 
104
 
105
+ @spaces.GPU(duration=150)
106
  def llama_generate(
107
  message: str,
108
  max_new_tokens: int = LLAMA_DEFAULT_MAX_NEW_TOKENS,