Skip to content

Commit c8f4b52

Browse files
committed
Add note
Signed-off-by: Rafael Vasquez <rafvasq21@gmail.com>
1 parent 2b5e960 commit c8f4b52

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/MODEL.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ export DEFAULT_TYPES="fp16"
9393
export DS_PATH=/resources/sharegpt/share_gpt.json
9494
```
9595

96-
Then, run the metrics generation script:
96+
Then, run the metrics generation script. The ShareGPT dataset will be automatically downloaded if it doesn't exist:
9797

9898
```bash
9999
python generate_metrics.py --architecture=hf_pretrained --model_path=$MODEL_PATH --tokenizer=$MODEL_PATH --unfuse_weights --output_dir=/tmp/aiu-fms-testing-utils/output/ --compile_dynamic --max_new_tokens=$MAX_NEW_TOKENS --min_pad_length=$SEQ_LENS --batch_size=$BATCH_SIZES --default_dtype=$DEFAULT_TYPES --sharegpt_path=$DS_PATH --num_test_tokens_per_sequence=1024

0 commit comments

Comments
 (0)