From 01f54841e9d8edb590de63f01e6cf6e26b0a54cb Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Thu, 17 Aug 2023 12:08:05 -0700
Subject: [PATCH] Update llama2-model.libsonnet

---
 tests/pytorch/nightly/llama2-model.libsonnet | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/tests/pytorch/nightly/llama2-model.libsonnet b/tests/pytorch/nightly/llama2-model.libsonnet
index 26855339e..c3b5f6616 100644
--- 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
       |||,
     },
   },