Update finqwen_inference.py

aeef6b6a · wanglch · 47566627 · aeef6b6a
Commit aeef6b6a authored Jul 04, 2024 by wanglch
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

finqwen_inference.py finqwen_inference.py +2 -2

No files found.
--- a/finqwen_inference.py
+++ b/finqwen_inference.py
@@ -2,7 +2,7 @@ from modelscope import AutoModelForCausalLM, AutoTokenizer
 from modelscope import GenerationConfig
 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'