From a844119335e51dfd59d757076407515194dd415e Mon Sep 17 00:00:00 2001
From: yolain <yolain@163.com>
Date: Mon, 30 Dec 2024 11:26:51 +0800
Subject: [PATCH] Change lark-parser dependency to lark

---
 py/libs/translate.py | 4 ++--
 requirements.txt     | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

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