NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-28 17:29:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

cleanup

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-02 22:59:32 -05:00
parent 54bd9a5ae0
commit 672cdbaea8
8 changed files with 360 additions and 820 deletions
Download patch file Download diff file Expand all files Collapse all files

6
example_trainer/config.py
View file

@ -92,6 +92,12 @@ class TrainingConfig(BaseModel):
vllm_gpu_memory_utilization: float = Field(
0.45, description="GPU memory utilization for vLLM server (0.0-1.0)"
)
max_model_len: int = Field(
4096, description="Maximum context length for vLLM server"
)
dtype: str = Field(
"bfloat16", description="Data type for model weights (bfloat16, float16, auto)"
)
# === Weights & Biases Configuration ===
use_wandb: bool = Field(