mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
34db804c76
* issue #76, load_checkpoint_and_dispatch() 'force_hooks' https://github.com/ostris/ai-toolkit/issues/76 * RunPod cloud config https://github.com/ostris/ai-toolkit/issues/90 * change 2x A40 to 1x A40 and price per hour referring to https://github.com/ostris/ai-toolkit/issues/90#issuecomment-2294894929 * include missed FLUX.1-schnell setup guide in last commit * huggingface-cli login required auth * #92 peft, #114 colab, schnell training in colab * modal cloud - run_modal.py and .yaml configs * run_modal.py mount path example * modal_examples renamed to modal * Training in Modal README.md setup guide * rename run command in title for consistency
32 lines
406 B
Plaintext
32 lines
406 B
Plaintext
torch
|
|
torchvision
|
|
safetensors
|
|
git+https://github.com/huggingface/diffusers.git
|
|
transformers
|
|
lycoris-lora==1.8.3
|
|
flatten_json
|
|
pyyaml
|
|
oyaml
|
|
tensorboard
|
|
kornia
|
|
invisible-watermark
|
|
einops
|
|
accelerate
|
|
toml
|
|
albumentations
|
|
pydantic
|
|
omegaconf
|
|
k-diffusion
|
|
open_clip_torch
|
|
timm
|
|
prodigyopt
|
|
controlnet_aux==0.0.7
|
|
python-dotenv
|
|
bitsandbytes
|
|
hf_transfer
|
|
lpips
|
|
pytorch_fid
|
|
optimum-quanto
|
|
sentencepiece
|
|
huggingface_hub
|
|
peft |