Typo Fix

Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>
pytorch · Sep 13, 2023 · 6dc025b · 6dc025b
1 parent 7174cde
commit 6dc025b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/cpp/src/examples/babyllama/baby_llama_handler.hh b/cpp/src/examples/babyllama/baby_llama_handler.hh
@@ -1,5 +1,5 @@
-#ifndef LLM_HANDLER_HH_
-#define LLM_HANDLER_HH_
+#ifndef BABYLLAMA_HANDLER_HH_
+#define BABYLLAMA_HANDLER_HH_
 
 #include <folly/FileUtil.h>
 #include <folly/json.h>
@@ -45,4 +45,4 @@ class BabyLlamaHandler : public torchserve::torchscripted::BaseHandler {
       override;
 };
 }  // namespace llm
-#endif  // LLM_HANDLER_HH_
+#endif  // BABYLLAMA_HANDLER_HH_
diff --git a/cpp/test/backends/torch_scripted/torch_scripted_backend_test.cc b/cpp/test/backends/torch_scripted/torch_scripted_backend_test.cc
@@ -78,7 +78,7 @@ TEST_F(TorchScriptedBackendTest, TestLoadPredictMnistHandler) {
                     "mnist_ts", 200);
 }
 
-TEST_F(TorchScriptedBackendTest, TestLoadPredictLlmHandler) {
+TEST_F(TorchScriptedBackendTest, TestLoadPredictBabyLlamaHandler) {
   this->LoadPredict(
       std::make_shared<torchserve::LoadModelRequest>(
           "test/resources/torchscript_model/babyllama/babyllama_handler", "llm",

diff --git a/cpp/test/resources/torchscript_model/babyllama/babyllama_handler/MAR-INF/MANIFEST.json b/cpp/test/resources/torchscript_model/babyllama/babyllama_handler/MAR-INF/MANIFEST.json
@@ -2,7 +2,7 @@
   "createdOn": "28/07/2020 06:32:08",
   "runtime": "LSP",
   "model": {
-    "modelName": "llm",
+    "modelName": "babyllama",
     "serializedFile": "dummy.pt",
     "handler": "libbabyllama_handler:BabyLlamaHandler",
     "modelVersion": "2.0"