clarke / scripts /train_lora.py

Commit History

  1. Fix: remove unsupported SFTConfig params

    9ffb169
  2. Fix: USER env for torch cache, remove duplicate import

    b64e033
  3. Fix: use SFTConfig for newer TRL API

    e6597b5
  4. Fix: remove deprecated dataset_text_field for newer TRL

    b9da45b
  5. Fix: tokenizer -> processing_class for newer TRL

    5321d11
  6. Fix: set USER env var for PyTorch cache in container

    09e0926
  7. Fix: total_mem -> total_memory in training script

    adbdf7d
  8. Add optional LoRA training on startup (RUN_LORA_TRAINING flag)

    7b3c958