Add assertions

Author: Lorenzo Toniazzi

tests/test_lora_conversion_and_inference.sh

@@ -1,6 +1,13 @@
 #!/bin/bash
 set -e
 
+# Array of models to iterate over
+declare -a params=(
+    # "Gemma2ForCausalLM 64"
+    "LlamaForCausalLM 64"
+    "Phi3ForCausalLM 64"
+)
+
 MODELS_REPO=reduce-llms-for-testing
 MODELS_REPO_URL=https://huggingface.co/ltoniazzi/$MODELS_REPO
 
@@ -108,19 +115,12 @@ run_conversion_and_inference_lora() {
     echo "All steps completed for $model_name with size $size_matrix!"
 }
 
-# Array of parameters to iterate over
-declare -a params=(
-    # "Gemma2ForCausalLM 64"
-    "LlamaForCausalLM 64"
-    "Phi3ForCausalLM 64"
-)
-
-# Loop through each set of parameters
+# Run test for each model
 for param in "${params[@]}"; do
     run_conversion_and_inference_lora $param
 done
 
-# Print all collected results after the loop completes
+# Print all collected results
 echo -e "\n\n\033[1mSummary of All Results:\033[0m"
 for result in "${results[@]}"; do
     echo -e "$result"