Spaces:
Sleeping
Sleeping
Commit
·
62ea137
1
Parent(s):
d62ccf6
bugfix: convert to int type for max_new_tokens
Browse files- env_config.py +1 -1
env_config.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
HF_API_TOKEN = os.environ.get("HF_API_TOKEN")
|
| 4 |
-
MAX_NEW_TOKENS = os.environ.get("MAX_NEW_TOKENS", 1024)
|
| 5 |
MODEL_ID = os.environ.get("MODEL_ID", "mistralai/Mistral-7B-Instruct-v0.2")
|
|
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
HF_API_TOKEN = os.environ.get("HF_API_TOKEN")
|
| 4 |
+
MAX_NEW_TOKENS = int(os.environ.get("MAX_NEW_TOKENS", 1024))
|
| 5 |
MODEL_ID = os.environ.get("MODEL_ID", "mistralai/Mistral-7B-Instruct-v0.2")
|