curl --request POST \
--url https://api.baseten.co/v1/training_projects/{training_project_id}/jobs \
--header "Authorization: Api-Key $BASETEN_API_KEY" \
--data '{
"training_job": {
"image": {
"base_image": "hello-world",
"docker_auth": null
},
"compute": {
"node_count": 1,
"cpu_count": 1,
"memory": "2Gi",
"accelerator": {
"accelerator": "H100",
"count": 2
}
},
"runtime": {
"start_commands": [
"python main.py"
],
"environment_variables": {
"API_KEY": "your_api_key_here",
"PATH": "/usr/bin"
},
"artifacts": null,
"enable_cache": true,
"cache_config": {
"enable_legacy_hf_mount": true,
"enabled": true,
"mount_base_path": "/root/.cache",
"require_cache_affinity": true
},
"checkpointing_config": {
"enabled": true,
"checkpoint_path": "/mnt/ckpts",
"volume_size_gib": 10
},
"load_checkpoint_config": null
},
"name": "gpt-oss-job",
"truss_user_env": null,
"interactive_session": null,
"weights": [
{
"allow_patterns": null,
"auth": null,
"auth_secret_name": null,
"ignore_patterns": null,
"mount_location": "/app/models/base",
"source": "hf://meta-llama/Llama-3-8B@main"
}
],
"enable_baseten_workdir": false,
"priority": 0
}
}'