Unverified Commit 22d403f5 authored by tpoisonooo's avatar tpoisonooo Committed by GitHub
Browse files

fix(project): interlm run error (#69)

parent 7c6edc83
.vscode/ .vscode/
.idea/ .idea/
.egg/
__pycache__/ __pycache__/
*.egg-info/ *.egg-info/
workspace/ workspace/
......
...@@ -62,7 +62,7 @@ Below are quick steps for installation: ...@@ -62,7 +62,7 @@ Below are quick steps for installation:
```shell ```shell
conda create -n lmdeploy python=3.10 conda create -n lmdeploy python=3.10
conda activate lmdeploy conda activate lmdeploy
git clone clone https://github.com/InternLM/lmdeploy.git git clone https://github.com/InternLM/lmdeploy.git
cd lmdeploy cd lmdeploy
pip install -e . pip install -e .
``` ```
......
...@@ -19,7 +19,7 @@ def main(model_name, model_path, session_id: int = 1): ...@@ -19,7 +19,7 @@ def main(model_name, model_path, session_id: int = 1):
tm_model = tm.TurboMind(model_path) tm_model = tm.TurboMind(model_path)
generator = tm_model.create_instance() generator = tm_model.create_instance()
tokenizer_model_path = osp.join(model_path, 'triton_models', 'tokenizer') tokenizer_model_path = osp.join(model_path, 'triton_models', 'tokenizer')
tokenizer = AutoTokenizer.from_pretrained(tokenizer_model_path) tokenizer = AutoTokenizer.from_pretrained(tokenizer_model_path, trust_remote_code=True)
model = MODELS.get(model_name)() model = MODELS.get(model_name)()
nth_round = 1 nth_round = 1
......
...@@ -5,3 +5,4 @@ safetensors ...@@ -5,3 +5,4 @@ safetensors
setuptools setuptools
transformers<4.30.0 transformers<4.30.0
tritonclient==2.33.0 tritonclient==2.33.0
pybind11
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment