meisaicheck-api / config.py
vumichien's picture
update GPU config
c41415e
import os
# Security Config
SECRET_KEY = "09d25e094faa6ca2556c818166b7a9563b93f7099f6f0f4caa6cf63b88e8d3e7"
ALGORITHM = "HS256"
ACCESS_TOKEN_EXPIRE_HOURS = 24
# Paths
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DATA_DIR = os.path.join(BASE_DIR, "data")
UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
OUTPUT_DIR = os.path.join(BASE_DIR, "outputs")
SUBJECT_DATA_FILE = os.path.join(DATA_DIR, "subjectData.csv")
SAMPLE_DATA_FILE = os.path.join(DATA_DIR, "sampleData.csv")
STANDARD_NAME_MAP_DATA_FILE = os.path.join(DATA_DIR, "standardData.csv")
# Model Config
HALF = "float16"
MODEL_TYPE = None
DEVICE_TYPE = "cpu"
MODEL_NAME = "Detomo/cl-nagoya-sup-simcse-ja-nss-v_1_0_6"
if HALF == "float16":
SENTENCE_EMBEDDING_FILE = os.path.join(
DATA_DIR,
"anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_6_float16.pkl",
)
elif HALF == "bfloat16":
SENTENCE_EMBEDDING_FILE = os.path.join(
DATA_DIR,
"anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_6_bfloat16.pkl",
)
else:
SENTENCE_EMBEDDING_FILE = os.path.join(
DATA_DIR,
"anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_6.pkl",
)
# SENTENCE_EMBEDDING_FILE = None