run_prompt_creation_dummy.sh 379 Bytes
Newer Older
sanchit-gandhi's avatar
sanchit-gandhi committed
1
2
3
#!/usr/bin/env bash

python run_prompt_creation.py \
sanchit-gandhi's avatar
sanchit-gandhi committed
4
5
  --dataset_name "ylacombe/libritts_r_tags_and_text" \
  --dataset_config_name "clean" \
sanchit-gandhi's avatar
sanchit-gandhi committed
6
  --dataset_split_name "dev.clean" \
sanchit-gandhi's avatar
sanchit-gandhi committed
7
  --model_name_or_path "mistralai/Mistral-7B-Instruct-v0.2" \
sanchit-gandhi's avatar
sanchit-gandhi committed
8
  --per_device_eval_batch_size 2 \
sanchit-gandhi's avatar
sanchit-gandhi committed
9
10
  --attn_implementation "sdpa" \
  --dataloader_num_workers 0 \
sanchit-gandhi's avatar
sanchit-gandhi committed
11
  --output_dir "./" \
sanchit-gandhi's avatar
sanchit-gandhi committed
12
  --load_in_4bit