chore: 更新脚本并且修改gitignore

scripts/demo/download.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 from huggingface_hub import snapshot_download
 
-PROJECT_ROOT = Path(__file__).parent.parent
+PROJECT_ROOT = Path(__file__).resolve().parents[2]
 PARAMETER_ROOT = Path(PROJECT_ROOT, "params")
 
 if __name__ == "__main__":

scripts/demo/generate_ar.py

@@ -3,7 +3,7 @@ from pathlib import Path
 from astrai.config.param_config import ModelParameter
 from astrai.inference.generator import GeneratorFactory, GenerationRequest
 
-PROJECT_ROOT = Path(__file__).parent.parent
+PROJECT_ROOT = Path(__file__).resolve().parents[2]
 PARAMETER_ROOT = Path(PROJECT_ROOT, "params")
 
 

scripts/demo/generate_batch.py

@@ -3,7 +3,7 @@ from pathlib import Path
 from astrai.config.param_config import ModelParameter
 from astrai.inference.generator import GeneratorFactory, GenerationRequest
 
-PROJECT_ROOT = Path(__file__).parent.parent
+PROJECT_ROOT = Path(__file__).resolve().parents[2]
 PARAMETER_ROOT = Path(PROJECT_ROOT, "params")
 
 

scripts/demo/stream_chat.py

@@ -3,7 +3,7 @@ from pathlib import Path
 from astrai.config.param_config import ModelParameter
 from astrai.inference.generator import GeneratorFactory, GenerationRequest
 
-PROJECT_ROOT = Path(__file__).parent.parent
+PROJECT_ROOT = Path(__file__).resolve().parents[2]
 PARAMETER_ROOT = Path(PROJECT_ROOT, "params")
 
 
@@ -25,6 +25,7 @@ def chat():
             max_len=param.config.max_len,
             history=history,
             system_prompt=None,
+            stream=True,
         )
         generator = GeneratorFactory.create(param, request)