annotation / mixtral_tune.sh
MudeHui's picture
Add application file
1fb65ae
raw
history blame contribute delete
501 Bytes
pip install --upgrade pip
pip install -q -U bitsandbytes
pip install -q -U git+https://github.com/huggingface/transformers.git
pip install -q -U git+https://github.com/huggingface/peft.git
pip install -q -U git+https://github.com/huggingface/accelerate.git
pip install -q -U datasets scipy ipywidgets matplotlib
train_json="./data/chat_edit/assets/test200/edit_instructions_v0.jsonl"
output_dir = f"{output_root}/{run_name}"
python3 ./dataset_creation/mixtral_tune.py \
--train_json train_json