Merge pull request #722 from relyt0925/fix-shm-size

Add environment variables for ilab container and increase shm size for vllm

training/ilab-wrapper/ilab

@@ -76,11 +76,15 @@ PODMAN_COMMAND=("sudo" "podman" "run" "--rm" "-it"
 	"${IMPERSONATE_CURRENT_USER_PODMAN_FLAGS[@]}"
 	"--device" "${CONTAINER_DEVICE}"
 	"--security-opt" "label=disable" "--net" "host"
+	"--shm-size" "10G"
 	"-v" "$HOME:$HOME"
 	"${ADDITIONAL_MOUNT_OPTIONS[@]}"
 	# This is intentionally NOT using "--env" "HOME" because we want the HOME
 	# of the current shell and not the HOME set by sudo
 	"--env" "HOME=$HOME"
+	"--env" "ILAB_GLOBAL_CONFIG=$ILAB_GLOBAL_CONFIG"
+	"--env" "VLLM_LOGGING_LEVEL=$VLLM_LOGGING_LEVEL"
+	"--env" "NCCL_DEBUG=$NCCL_DEBUG"
 	"--entrypoint" "$ENTRYPOINT"
 	"--env" "HF_TOKEN"
 	"${IMAGE_NAME}")