diff --git a/training/ilab-wrapper/ilab b/training/ilab-wrapper/ilab index eb42e7e0..d49dea6f 100755 --- a/training/ilab-wrapper/ilab +++ b/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}")