Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
|
|
9 |
|
10 |
MAX_MAX_NEW_TOKENS = 100
|
11 |
DEFAULT_MAX_NEW_TOKENS = 20
|
12 |
-
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "
|
13 |
|
14 |
DESCRIPTION = """\
|
15 |
# Löwolf Chat
|
@@ -37,7 +37,7 @@ def generate(
|
|
37 |
message: str,
|
38 |
chat_history: list[tuple[str, str]],
|
39 |
system_prompt: str,
|
40 |
-
max_new_tokens: int =
|
41 |
temperature: float = 0.6,
|
42 |
top_p: float = 0.9,
|
43 |
top_k: int = 50,
|
|
|
9 |
|
10 |
MAX_MAX_NEW_TOKENS = 100
|
11 |
DEFAULT_MAX_NEW_TOKENS = 20
|
12 |
+
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "200"))
|
13 |
|
14 |
DESCRIPTION = """\
|
15 |
# Löwolf Chat
|
|
|
37 |
message: str,
|
38 |
chat_history: list[tuple[str, str]],
|
39 |
system_prompt: str,
|
40 |
+
max_new_tokens: int = 50,
|
41 |
temperature: float = 0.6,
|
42 |
top_p: float = 0.9,
|
43 |
top_k: int = 50,
|