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

sanchit-gandhi's avatar
sanchit-gandhi committed
3
accelerate launch --multi_gpu --mixed_precision=fp16 --num_processes=8 run_prompt_creation.py \
sanchit-gandhi's avatar
sanchit-gandhi committed
4
  --dataset_name "parler-tts/libritts-r-tags-and-text" \
sanchit-gandhi's avatar
sanchit-gandhi committed
5
  --dataset_config_name "clean" \
sanchit-gandhi's avatar
sanchit-gandhi committed
6
  --model_name_or_path "mistralai/Mistral-7B-Instruct-v0.2" \
sanchit-gandhi's avatar
sanchit-gandhi committed
7
8
  --per_device_eval_batch_size 64 \
  --attn_implementation "sdpa" \
sanchit-gandhi's avatar
sanchit-gandhi committed
9
10
11
12
  --dataloader_num_workers 4 \
  --output_dir "./" \
  --load_in_4bit \
  --push_to_hub \
sanchit-gandhi's avatar
sanchit-gandhi committed
13
  --hub_dataset_id "parler-tts/libritts-r-tags-and-text-generated"
sanchit-gandhi's avatar
sanchit-gandhi committed
14
15

accelerate launch --multi_gpu --mixed_precision=fp16 --num_processes=8 run_prompt_creation.py \
sanchit-gandhi's avatar
sanchit-gandhi committed
16
  --dataset_name "parler-tts/libritts-r-tags-and-text" \
sanchit-gandhi's avatar
sanchit-gandhi committed
17
18
19
20
21
22
23
24
  --dataset_config_name "other" \
  --model_name_or_path "mistralai/Mistral-7B-Instruct-v0.2" \
  --per_device_eval_batch_size 64 \
  --attn_implementation "sdpa" \
  --dataloader_num_workers 4 \
  --output_dir "./" \
  --load_in_4bit \
  --push_to_hub \
sanchit-gandhi's avatar
sanchit-gandhi committed
25
  --hub_dataset_id "parler-tts/libritts-r-tags-and-text-generated"