Sam-Oliveira
fix triton cache directory
7ce7422
raw
history blame contribute delete
361 Bytes
from pathlib import Path
import pathlib,tempfile
# Root folder for DB
#PROJ = Path(__file__).parent # For MAC
PROJ = pathlib.Path(tempfile.gettempdir()) # For Space
MAX_RESULTS = 10 #default number of results
MODEL_NAME = "unsloth/llama-3-8b-Instruct-bnb-4bit" #default model
DB_FILE = "papers.db" # default database file
DB_PATH = PROJ / DB_FILE