[frontend][LLM] Change vocab path

buddy-compiler · Oct 25, 2023 · d880ffa · d880ffa
1 parent b139a31
commit d880ffa
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/Interface/core/TextContainerTest.cpp b/tests/Interface/core/TextContainerTest.cpp
@@ -34,8 +34,7 @@ int main() {
   //
   // The test running directory is in <build dir>/tests/Interface/core, so the
   // vocabulary directory uses the following relative path.
-  std::string vocabDir = "/root/buddy-mlir/tests/Interface/core/vocab_bert.txt";
-  // std::string vocabDir = "../../../../tests/Interface/core/vocab_bert.txt";
+  std::string vocabDir = "../../../../tests/Interface/core/vocab_bert.txt";
   //===--------------------------------------------------------------------===//
   // Test text constructor for pure string.
   //===--------------------------------------------------------------------===//
@@ -227,8 +226,7 @@ int main() {
   //
   // The test running directory is in <build dir>/tests/Interface/core, so the
   // vocabulary directory uses the following relative path.
-  vocabDir = "/root/buddy-mlir/tests/Interface/core/vocab_llama.txt";
-  //vocabDir = "../../../../tests/Interface/core/vocab_llama.txt";
+  vocabDir = "../../../../tests/Interface/core/vocab_llama.txt";
   //===--------------------------------------------------------------------===//
   // Test text constructor for pure string using Llama tokenizer.
   //===--------------------------------------------------------------------===//