File size: 523 Bytes
f0e4d3b
 
 
 
 
 
 
 
 
5c5f097
f0e4d3b
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import os

MONGO_DB_URI = os.environ.get("MONGO_DB_URI", "mongodb://localhost:27017")
MONGO_DB_NAME = os.environ.get("MONGO_DB_NAME", "test_db")
LANGUAGE = "ja"

DEBUG = os.getenv("DEBUG", "False").lower() in ["true", "1", "yes"]
LOCAL_TESTING = os.getenv("LOCAL_TESTING", "False").lower() in ["true", "1", "yes"]
MODEL_SELECTION_MODE = os.getenv("MODEL_SELECTION_MODE", "random")
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "512"))

if LOCAL_TESTING:
  MAX_NEW_TOKENS = 20
else:
  MAX_NEW_TOKENS = 512