clarke / scripts /train_lora.py

Commit History

Fix: remove unsupported SFTConfig params
9ffb169

yashvshetty commited on

Fix: USER env for torch cache, remove duplicate import
b64e033

yashvshetty commited on

Fix: use SFTConfig for newer TRL API
e6597b5

yashvshetty commited on

Fix: remove deprecated dataset_text_field for newer TRL
b9da45b

yashvshetty commited on

Fix: tokenizer -> processing_class for newer TRL
5321d11

yashvshetty commited on

Fix: set USER env var for PyTorch cache in container
09e0926

yashvshetty commited on

Fix: total_mem -> total_memory in training script
adbdf7d

yashvshetty commited on

Add optional LoRA training on startup (RUN_LORA_TRAINING flag)
7b3c958

yashvshetty commited on