File size: 204 Bytes
8177cf7 |
1 2 3 4 5 6 7 8 9 |
#!/usr/bin/env bash
python3 run_initialization.py \
--model_name_or_path "meta-llama/Meta-Llama-3.1-8B" \
--num_hidden_layers "6" \
--output_dir "./" \
--torch_dtype "bfloat16" \
--push_to_hub
|