Add LoRA fine-tuning scripts (#5)

2025-09-16 21:03:14 -07:00
parent d244401c9d
commit ed52bd539b
8 changed files with 1372 additions and 46 deletions
@@ -20,6 +20,8 @@ dependencies = [
    "torch",
    "accelerate==1.6.0",
    "transformers==4.51.3", # we develop this project on transformers==4.51.3, later version may not be compatible
+    "datasets==3.5.0", # avoid issues with torchcodec
+    "peft", # for finetuning
    "llvmlite>=0.40.0",
    "numba>=0.57.0",
    "diffusers",