diff --git a/py/libs/translate.py b/py/libs/translate.py
index 577c92b..89e304d 100644
--- a/py/libs/translate.py
+++ b/py/libs/translate.py
@@ -12,8 +12,8 @@ from .utils import install_package
 try:
     from lark import Lark, Transformer, v_args
 except:
-    print('install lark-parser...')
-    install_package('lark-parser')
+    print('install lark...')
+    install_package('lark')
     from lark import Lark, Transformer, v_args
 
 model_path = os.path.join(folder_paths.models_dir, 'prompt_generator')
diff --git a/requirements.txt b/requirements.txt
index 3b0bdde..29d2b5d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 diffusers
 accelerate
 clip_interrogator>=0.6.0
-lark-parser
+lark
 onnxruntime
 opencv-python
 sentencepiece