From b30bea325738065411660bea37fcb91c03b6a594 Mon Sep 17 00:00:00 2001
From: ngxson <thichthat@gmail.com>
Date: Fri, 24 May 2024 22:50:03 +0200
Subject: [PATCH] add comments

---
 .../control-vector-generator/control-vector-generator.cpp  | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/examples/control-vector-generator/control-vector-generator.cpp b/examples/control-vector-generator/control-vector-generator.cpp
index 2195e28fa..e72960636 100644
--- a/examples/control-vector-generator/control-vector-generator.cpp
+++ b/examples/control-vector-generator/control-vector-generator.cpp
@@ -13,9 +13,10 @@ struct callback_data {
     int n_tokens = 0;
     int n_embd = 0;
     bool is_eval_pos = true;
-    std::vector<float *> v_pos;
-    std::vector<float *> v_neg;
-    std::vector<float *> v_diff;
+    // each element of the vector correspond to one layer
+    std::vector<float *> v_pos;  // vector of matrices of size [n_embd, n_tokens]
+    std::vector<float *> v_neg;  // vector of matrices of size [n_embd, n_tokens]
+    std::vector<float *> v_diff; // vector of matrices of size [n_embd, n_tokens]
 };
 
 static std::string ggml_ne_string(const ggml_tensor * t) {