Skip to content

Commit a9d1d49

Browse files
authored
Merge pull request #28 from RICHARDNAN/patch-2
Add RANK and LOCAL_WORLD_SIZE environment variables
2 parents fce8d8a + 376e9d6 commit a9d1d49

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

doc/zh/DeepseekR1_tutorial_zh_for_Ascend_NPU.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,8 @@ export TASK_QUEUE_ENABLE=0 # 保证算子下发顺序有序
108108
export USE_MERGE=0
109109
export INF_NAN_MODE_FORCE_DISABLE=1
110110
export TASK_QUEUE_ENABLE=0
111+
export RANK=0
112+
export LOCAL_WORLD_SIZE=1
111113
#export PROF_DECODE=1
112114
#export PROF_PREFILL=1
113115

@@ -116,7 +118,6 @@ source /usr/local/Ascend/nnal/atb/set_env.sh
116118

117119
python ktransformers/server/main.py \
118120
--port 10002 \
119-
120121
--model_path /mnt/data/models/DeepSeek-R1-q4km-w8a8 \
121122
--gguf_path /mnt/data/models/DeepSeek-R1-q4km-w8a8 \
122123
--model_name DeepSeekV3ForCausalLM \

0 commit comments

Comments
 (0)