Files
ai-toolkit/toolkit/assistant_lora.py