add yamlinclude package

Dockerfile

@@ -6,7 +6,7 @@ RUN apt-get -y update \
     && rm -rf /var/lib/apt/lists/*
 
 RUN python3 -m pip install --upgrade pip \
-    && pip install --no-cache-dir  torch torchvision torchaudio
+    && pip install --no-cache-dir  torch torchvision torchaudio yamlinclude
 
 RUN useradd -m -u 1000 user
 

startup.sh

@@ -6,7 +6,7 @@ echo "PWD: $(pwd)"
 echo $HF_TOKEN > .hf_token
 echo "LS: $(ls -als)"
 
-while true; do nvidia-smi; sleep 600; done &
+while true; do nvidia-smi; sleep 60; done &
 
 python3 -c "import torch; \
     print(f\"is availeble = {torch.cuda.is_available()}\"); \