Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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,
|