Skip to content

GitLab

Commit aeef6b6a authored by wanglch's avatar wanglch
Browse files

Update finqwen_inference.py

parent 47566627
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
finqwen_inference.py
View file @ aeef6b6a
...@@ -2,7 +2,7 @@ from modelscope import AutoModelForCausalLM, AutoTokenizer ...@@ -2,7 +2,7 @@ from modelscope import AutoModelForCausalLM, AutoTokenizer
from modelscope import GenerationConfig from modelscope import GenerationConfig
import os import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3" os.environ["HIP_VISIBLE_DEVICES"] = "0,1,2,3"
model_dir = '/home/wanglch/projects/FinQwen/Tongyi-Finance-14B-Chat' model_dir = '/home/wanglch/projects/FinQwen/Tongyi-Finance-14B-Chat'
......
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