print

Signed-off-by: Will Johnson <mwjohnson728@gmail.com>

Author: willmj

build/Dockerfile

@@ -149,6 +149,7 @@ RUN --mount=type=cache,target=/home/${USER}/.cache/pip,uid=${USER_UID} \
     python -m pip install --user wheel && \
     python -m pip install --user "$(head bdist_name)" && \
     python -m pip install --user "$(head bdist_name)[flash-attn]" && \
+    python -c "import torch; print(torch.__version__)" && \
     python -m pip install --user "$(head bdist_name)[mamba]"
 
 # fms_acceleration_peft = PEFT-training, e.g., 4bit QLoRA