Collect results for different models before print

Author: Lorenzo Toniazzi

tests/test_lora_conversion_and_inference.sh

@@ -11,6 +11,8 @@ else
     echo "Repository already exists. Skipping clone."
 fi
 
+# Declare a regular array to store results
+results=()
 
 run_conversion_and_inference_lora() {
     local model_name=$1
@@ -46,23 +48,31 @@ run_conversion_and_inference_lora() {
     OUTPUT_LORA_MERGED=$(llama-cli -m reduce-llms-for-testing/$model_name/size=$size_matrix/base/Base-$model_size_mb-F32-lora-merged.gguf \
     -p "<bos>I see a little silhouetto" -n 50 --seed 42)
 
-
-    # Echo the outputs with bullet points and spacing
-    echo -e "\n\n\n\033[1mResults:\033[0m"
-    echo -e "\n  • \033[32mBase:\n $OUTPUT_BASE"  # Green color for "BASE"
-    echo -e "\n  • \033[34mLora hot:\n $OUTPUT_LORA_HOT"  # Blue color for "Lora hot"
-    echo -e "\n  • \033[36mLora merged:\n $OUTPUT_LORA_MERGED"  # Cyan color for "Lora merged"
-    echo -e "\n\n\n \033[0m"
-
+    # Store the results in the regular array
+    results+=("
+    \n\n\n\033[1mResults for $model_name with size $size_matrix and model size $model_size_mb:\033[0m
+    \n  • \033[32mBase:\n $OUTPUT_BASE
+    \n  • \033[34mLora hot:\n $OUTPUT_LORA_HOT
+    \n  • \033[36mLora merged:\n $OUTPUT_LORA_MERGED
+    \n\n\n \033[0m
+    ")
 
     echo "All steps completed for $model_name with size $size_matrix and model size $model_size_mb!"
 }
 
+# Array of parameters to iterate over
 declare -a params=(
     "Gemma2ForCausalLM 64 19M"
     # "AnotherModel 128 25M"
 )
 
+# Loop through each set of parameters
 for param in "${params[@]}"; do
     run_conversion_and_inference_lora $param
-done
+done
+
+# Print all collected results after the loop completes
+echo -e "\n\n\033[1mSummary of All Results:\033[0m"
+for result in "${results[@]}"; do
+    echo -e "$result"
+done