Typo to open PR

huggingface · Jan 30, 2024 · 8b79ee2 · 8b79ee2
1 parent 0cf83ce
commit 8b79ee2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -19,7 +19,7 @@
 
 
 if not is_accelerate_available() and not is_tgi_available():
-    hlog_warn("Using either accelerate or text-generation to run this script is advised.")
+    hlog_warn("Using either accelerate or text-generation-inference to run this script is advised.")
 
 TOKEN = os.getenv("HF_TOKEN")
 CACHE_DIR = os.getenv("HF_HOME")