| # config.py | |
| # ============================== | |
| # Central configuration file | |
| # After training, ONLY THIS FILE needs small changes | |
| # ============================== | |
| # Base model (always loaded) | |
| BASE_MODEL = "Qwen/Qwen2-0.5B-Instruct" | |
| # Core adapter (enable AFTER core training) | |
| #CORE_ADAPTER = None | |
| CORE_ADAPTER = "himu1780/DocuMint-Core" # ← uncomment after core training | |
| # Skill adapters (enable AFTER skill training) | |
| SKILL_ADAPTERS = { | |
| # "vat": "himu1780/DocuMint-Skill-vat", | |
| # "docs": "himu1780/DocuMint-Skill-docs", | |
| # "excel": "himu1780/DocuMint-Skill-excel", | |
| } | |
| # Generation defaults | |
| MAX_NEW_TOKENS = 256 | |
| TEMPERATURE = 0.4 | |
| TOP_P = 0.9 | |