Updated QwQ train notebook

This commit is contained in:
2025-03-29 11:03:53 +00:00
parent 585f655ad3
commit 30eb5c419d
3 changed files with 83 additions and 198 deletions

View File

@@ -12,10 +12,10 @@ export MSYS_NO_PATHCONV=1
# Pod configuration
GPU_TYPE=${GPU_TYPE:-"NVIDIA RTX 6000 Ada Generation"}
GPU_COUNT=${GPU_COUNT:-1}
CONTAINER_DISK_SIZE=${CONTAINER_DISK_SIZE:-100} # GB
CONTAINER_DISK_SIZE=${CONTAINER_DISK_SIZE:-200} # GB
CPU_COUNT=${CPU_COUNT:-1} # vCPUs
POD_NAME=${POD_NAME:-"sia-agent"}
VOLUME_SIZE=${VOLUME_SIZE:-200} # GB
VOLUME_SIZE=${VOLUME_SIZE:-50} # GB
VOLUME_PATH=${VOLUME_PATH:-"/root/data"} # Mount path within container
# Docker configuration