use pytorch container

Signed-off-by: Maanu Grover <maanug@nvidia.com>

Author: maanug-nv

.github/workflows/cache-hf-model.yml

@@ -29,6 +29,8 @@ jobs:
     name: Download Model
     runs-on: self-hosted-nemo
     environment: nemo-ci
+    container:
+      image: nvcr.io/nvidia/pytorch:25.05-py3
     env:
       HF_HUB_OFFLINE: 0
       HF_HOME: /mnt/datadrive/TestData/HF_HOME