junaid0600 commited on
Commit
c305f9f
Β·
verified Β·
1 Parent(s): 1867b72

Update training/train_agent.py

Browse files
Files changed (1) hide show
  1. training/train_agent.py +1 -1
training/train_agent.py CHANGED
@@ -39,7 +39,7 @@ from env.db_simulator import DatabaseSimulator
39
  # ── Config ────────────────────────────────────────────────────
40
  ENV_URL = os.getenv("ENV_URL", "https://junaid0600-sql-db-engineer-agent.hf.space")
41
  HF_TOKEN = os.getenv("HF_TOKEN", "")
42
- MODEL_NAME = os.getenv("MODEL_NAME", "unsloth/Qwen2.5-1.5B-Instruct")
43
  OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./sdea-trained")
44
  MAX_STEPS = int(os.getenv("MAX_STEPS", "100"))
45
 
 
39
  # ── Config ────────────────────────────────────────────────────
40
  ENV_URL = os.getenv("ENV_URL", "https://junaid0600-sql-db-engineer-agent.hf.space")
41
  HF_TOKEN = os.getenv("HF_TOKEN", "")
42
+ MODEL_NAME = os.getenv("MODEL_NAME", "unsloth/Qwen2.5-7B-Instruct")
43
  OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./sdea-trained")
44
  MAX_STEPS = int(os.getenv("MAX_STEPS", "100"))
45