Spaces:
Sleeping
Sleeping
Update training/train_agent.py
Browse files- 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-
|
| 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 |
|