Skip to content

Commit f9597fb

Browse files
authored
Make hf and converted model paths configurable via env vars with default values (#8)
Signed-off-by: Wang, Kai Lawrence <[email protected]>
1 parent a0c5fe6 commit f9597fb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/convert_block_fp8_to_channel_fp8.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@
22

33
pip install compress_pickle
44

5-
original_model_path=/models/DeepSeek-R1
6-
output_model_root=/models
5+
original_model_path=${ORIGINAL_MODEL_PATH:-/models/DeepSeek-R1}
6+
output_model_root=${OUTPUT_MODEL_ROOT:-/models}
77
input_scales_path=DeepSeek-R1-BF16-w8afp8-static-no-ste_input_scale_inv.pkl.gz
88

99
device=$(hl-smi -Q name -f csv | tail -n 1)

0 commit comments

Comments
 (0)