Update llama2-model.libsonnet

GoogleCloudPlatform · Aug 17, 2023 · 01f5484 · 01f5484
1 parent 4a8c9f6
commit 01f5484
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/pytorch/nightly/llama2-model.libsonnet b/tests/pytorch/nightly/llama2-model.libsonnet
@@ -146,14 +146,31 @@ local utils = import 'templates/utils.libsonnet';
     modelName+: '-n-i',
     tpuSettings+: {
       tpuVmExtraSetup: |||
+        # show current path
+        pwd
+        ls
         git clone -b llama2-google-next-inference https://github.com/pytorch-tpu/llama.git
         cd llama
+        # show current path
+        pwd
+        ls
+        pip list | grep torch
         pip install -r requirements.txt
         pip install -e .
+        pip list | grep torch
         # prepare data
+        # show current path
+        pwd
+        ls
         wget https://storage.mtls.cloud.google.com/manfei_bucket/LLaMA2/llama_2_model.zip
+        # show current path
+        pwd
+        ls
         sudo apt-get install unzip
         unzip llama_2_model.zip
+        # show current path
+        pwd
+        ls
       |||,
     },
   },