Spaces:
Running
Running
File size: 772 Bytes
e2b8b61 edf8cae e2b8b61 edf8cae e2b8b61 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | from pathlib import Path
from src.model_registry import DEFAULT_MODEL_ALIAS
BASE_DIR = Path(__file__).resolve().parent.parent
DATA_DIR = BASE_DIR / "data"
RAW_DATA_PATH = DATA_DIR / "raw" / "dailydialog.json"
ENCODED_DIR = DATA_DIR / "full" / "encoded"
LOCALIZED_DIR = DATA_DIR / "full" / "localized"
REVIEWED_DIR = DATA_DIR / "reviewed"
DECODED_DIR = DATA_DIR / "full" / "decoded"
EVAL_DIR = DATA_DIR / "evaluation"
PROMPTS_DIR = BASE_DIR / "prompts"
# =========================
# MODEL SETTINGS
# =========================
DEFAULT_MODEL = DEFAULT_MODEL_ALIAS
TEMPERATURE = 0.2
MAX_TOKENS = 4096
MAX_TOKENS_THINKING = 16384 # Higher budget for models that output thinking text
# =========================
# PIPELINE
# =========================
MAX_RETRIES = 2 |