We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a0c5fe6 commit f9597fbCopy full SHA for f9597fb
scripts/convert_block_fp8_to_channel_fp8.sh
@@ -2,8 +2,8 @@
2
3
pip install compress_pickle
4
5
-original_model_path=/models/DeepSeek-R1
6
-output_model_root=/models
+original_model_path=${ORIGINAL_MODEL_PATH:-/models/DeepSeek-R1}
+output_model_root=${OUTPUT_MODEL_ROOT:-/models}
7
input_scales_path=DeepSeek-R1-BF16-w8afp8-static-no-ste_input_scale_inv.pkl.gz
8
9
device=$(hl-smi -Q name -f csv | tail -n 1)
0 commit comments