mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 08:29:45 +00:00
* WIP, adding support for LTX2 * Training on images working * Fix loading comfy models * Handle converting and deconverting lora so it matches original format * Reworked ui to habdle ltx and propert dataset default overwriting. * Update the way lokr saves to it is more compatable with comfy * Audio loading and synchronization/resampling is working * Add audio to training. Does it work? Maybe, still testing. * Fixed fps default issue for sound * Have ui set fps for accurate audio mapping on ltx * Added audio procession options to the ui for ltx * Clean up requirements
41 lines
616 B
Plaintext
41 lines
616 B
Plaintext
torchao==0.10.0
|
|
safetensors
|
|
git+https://github.com/huggingface/diffusers@8600b4c10d67b0ce200f664204358747bd53c775
|
|
transformers==4.57.3
|
|
lycoris-lora==1.8.3
|
|
flatten_json
|
|
pyyaml
|
|
oyaml
|
|
tensorboard
|
|
kornia
|
|
invisible-watermark
|
|
einops
|
|
accelerate
|
|
toml
|
|
albumentations==1.4.15
|
|
albucore==0.0.16
|
|
pydantic
|
|
omegaconf
|
|
k-diffusion
|
|
open_clip_torch
|
|
timm
|
|
prodigyopt
|
|
controlnet_aux==0.0.10
|
|
python-dotenv
|
|
bitsandbytes
|
|
hf_transfer
|
|
lpips
|
|
pytorch_fid
|
|
optimum-quanto==0.2.4
|
|
sentencepiece
|
|
huggingface_hub
|
|
peft
|
|
gradio
|
|
python-slugify
|
|
opencv-python
|
|
pytorch-wavelets==1.3.0
|
|
matplotlib==3.10.1
|
|
setuptools==69.5.1
|
|
scipy==1.12.0
|
|
av==16.0.1
|
|
torchcodec |