github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Commit Graph

  • b4f0efb025 Version Bump Jaret Burkett 2026-04-28 13:42:00 -06:00
  • af6458d1b5 Enable caching of ACE step latents. Jaret Burkett 2026-04-28 13:39:20 -06:00
  • 77b8765939 Fix issue with hover styling on select components. Jaret Burkett 2026-04-28 10:55:20 -06:00
  • 43989cc19e Add advanced config section to the captioner Jaret Burkett 2026-04-28 10:49:46 -06:00
  • f972b750e6 Performance improvements to captioner. Add ability to set a default caption for ace captioner to avoid needing to caption the audio so we only transcribe Jaret Burkett 2026-04-28 10:15:11 -06:00
  • acc6a36214 Scale DFE 9 to a velocity equiv weight to match flow matching gradient strength. Probably need to rework all DFEs to do this as the math checks out. Jaret Burkett 2026-04-28 09:10:02 -06:00
  • 1fc4ad3979 Add sapiens2 as a diffusion feature extractor Jaret Burkett 2026-04-27 15:59:03 -06:00
  • 67d67f8c1d Ignore hidden files when captioning Jaret Burkett 2026-04-25 17:38:31 -06:00
  • 998a02f30e Yet another pseudo_huber fix. Still drinking coffee. Jaret Burkett 2026-04-19 10:05:40 -06:00
  • fc85410c9a Fix issue with precision on pseudo_huber loss Jaret Burkett 2026-04-19 10:02:27 -06:00
  • 20a99258b8 Fix issue is is else on pseudo_huber loss Jaret Burkett 2026-04-19 09:59:18 -06:00
  • f4445cd78c Added psuedo_huber loss Jaret Burkett 2026-04-19 09:51:46 -06:00
  • 488878f354 Use hidden layers in the loss for DFE 7 and 8 Jaret Burkett 2026-04-18 13:07:38 -06:00
  • beb40ae29b Add DFE8 with partial step Jaret Burkett 2026-04-17 17:40:16 -06:00
  • 7c4f18ce51 Fix ernie unpatchify Jaret Burkett 2026-04-17 12:03:39 -06:00
  • 8cb9649382 Add decode latent to ernie pipe Jaret Burkett 2026-04-17 12:01:09 -06:00
  • 67048df9f9 Merge branch 'main' of github.com:ostris/ai-toolkit Jaret Burkett 2026-04-17 06:05:40 -06:00
  • be54094704 Version bump Jaret Burkett 2026-04-17 06:05:26 -06:00
  • a513a1583e Fixed issue where Qwen VL MOE captioner produced nonsense Jaret Burkett 2026-04-16 21:24:43 +00:00
  • 22ea3dd620 Fixed issue on some systems where Logger didnt have atty Jaret Burkett 2026-04-16 21:09:52 +00:00
  • ab1ee4df34 Hotfix some issues with Wan models caused by diffusers and transformers updates Jaret Burkett 2026-04-16 20:53:50 +00:00
  • 0c18b39346 Version bump Jaret Burkett 2026-04-16 13:09:45 -06:00
  • afb62b1fa5 Add support for Nucleus-Image Jaret Burkett 2026-04-16 13:09:10 -06:00
  • 2faba22b46 Fix issue when saving advanced prompt embeds. No such file or directory error Jaret Burkett 2026-04-16 12:22:56 -06:00
  • 0792352dab version bump Jaret Burkett 2026-04-16 09:25:14 -06:00
  • 8f67f5022e Version lock peft version Jaret Burkett 2026-04-15 14:17:26 -06:00
  • acc3e60140 Add ernie to readme Jaret Burkett 2026-04-15 06:43:36 -06:00
  • dd7074a21f Fix issue with layer offloading on ernie Jaret Burkett 2026-04-14 19:19:02 -06:00
  • e74bc9ac7b Fix issue with concatinating advanced prompt embeds. Jaret Burkett 2026-04-14 16:04:34 -06:00
  • 7eb1226a6d Fix issue with loading advanced prompt configs metadata Jaret Burkett 2026-04-14 15:42:09 -06:00
  • 97d8c05d75 Fix issue with captioner sometimes outputting not utf-8 characters Jaret Burkett 2026-04-14 14:37:21 -06:00
  • 3e0c904054 Add support for Baidu's ERNIE-Image (#793) Jaret Burkett (Ostris) 2026-04-14 09:45:12 -06:00
  • e868fca562 fix custom_flowmatch_sampler (#783) Asaf Agami 2026-04-13 18:42:05 +03:00
  • 233e292256 Added some experimental low step things for zeta Jaret Burkett 2026-04-13 09:37:34 -06:00
  • 1058ef3513 Made AdvancedPromptEmbeds that is compatable with previous PromptEmbeds functionality, but is more streamlined and can accomidate more model embedding paradigms. Jaret Burkett 2026-04-11 10:45:47 -06:00
  • 0d11be41fa Adjust default sample steps. Jaret Burkett 2026-04-10 17:46:34 -06:00
  • 62e18427b4 Adjust default sample steps. Jaret Burkett 2026-04-10 17:46:09 -06:00
  • 9b4e2d1b0b More flac support Jaret Burkett 2026-04-10 12:27:09 -06:00
  • 0b9c365acb Add flac and ogg support Jaret Burkett 2026-04-10 12:10:54 -06:00
  • bfb373c8fa Prep for future breaking changes in newer versions of transformers library Jaret Burkett 2026-04-10 12:04:32 -06:00
  • 145144eee3 Fix issue with auto updating captions when captioning a dataset Jaret Burkett 2026-04-10 11:51:18 -06:00
  • 765a9d5b2e Add a download button to music samples in in the gear menu Jaret Burkett 2026-04-10 10:07:45 -06:00
  • d08ea8318f Change default timestep type for ace step to linear Jaret Burkett 2026-04-09 17:39:00 -06:00
  • 78cf049c29 Add support for ACE-Step 1.5 and ACE-Step 1.5 XL. Also added dataset captioning through the UI. (#785) Jaret Burkett 2026-04-09 15:02:03 -06:00
  • 9ca58e9aa2 Fixed offload and quantize order of ltx 2.3 text encoder. Jaret Burkett 2026-04-07 15:11:50 -06:00
  • 0dcbabf6af Fix merge nertwork ref Jaret Burkett 2026-04-01 10:38:31 -06:00
  • f213e3b1e5 Fix FLUX2 Klein load-time VRAM spikes on low-memory GPUs. (#726) M. Hofer 2026-04-01 17:36:55 +02:00
  • da2a79590f Add a merge network on save strength Jaret Burkett 2026-04-01 09:21:08 -06:00
  • 853ffaf207 Add light mode support. Jaret Burkett 2026-03-31 16:54:55 -06:00
  • ad474e3d06 Update and reformat the readme Jaret Burkett 2026-03-31 12:31:05 -06:00
  • 4a3251640a More work on compiling models Jaret Burkett 2026-03-31 12:11:56 -06:00
  • 358d684f6f Move compiiling the model after accelerate manipulation Jaret Burkett 2026-03-31 09:52:27 -06:00
  • 0045260af7 Fix issue where compile true did not actually compile the model Jaret Burkett 2026-03-31 09:27:54 -06:00
  • e22039e4aa Add more optimizers to the ui Jaret Burkett 2026-03-31 09:20:20 -06:00
  • bf56217c37 Fixed issue where job would fail if DB is locked. Jaret Burkett 2026-03-31 09:10:33 -06:00
  • dcb7f465ec Version Bump Jaret Burkett 2026-03-30 15:57:40 -06:00
  • 626d9674ea Add info about automated coding agent pull requests. So sick of them. Jaret Burkett 2026-03-30 11:02:12 -06:00
  • b43ea6c2d3 Abort caption requests when they are not in view to tax the server less. Jaret Burkett 2026-03-30 10:46:25 -06:00
  • a484e55d66 Rework dataset model and file dragging. Use single model for dragging, uploading, and selecting images. Jaret Burkett 2026-03-30 10:33:10 -06:00
  • ac82ebd852 Dont try to list hidden files in datasets Jaret Burkett 2026-03-30 10:03:10 -06:00
  • 171535833a Add Mac OS support for Apple Silicon (#770) Jaret Burkett 2026-03-30 09:37:47 -06:00
  • bc47fd6755 Make a requirements base file to make it easier to maintain requirements across platforms. Jaret Burkett 2026-03-29 14:04:46 -06:00
  • fbda10d088 Add a duplicate dataset function to the ui Jaret Burkett 2026-03-29 13:51:18 -06:00
  • 86dcf39eee Allow user to set a training seed via env vars for repeat result testing Jaret Burkett 2026-03-29 13:34:46 -06:00
  • 45e99664b9 Add icons to the top bar on the job page Jaret Burkett 2026-03-29 12:38:47 -06:00
  • 540659709d Improved the load time of dataset and sample images and videos by switching to streaming Jaret Burkett 2026-03-29 10:38:34 -06:00
  • e030f4f2e0 Show the control images in the image viewer when clicked so they can be easily previewed for reference. Jaret Burkett 2026-03-29 10:00:54 -06:00
  • affa411edc Fixed an issue where Flux.2 model VAE can be left offloaded to CPU when encoding control images while caching latents Jaret Burkett 2026-03-29 09:49:10 -06:00
  • 6a1fc54779 Add t0 loss target Jaret Burkett 2026-03-28 13:35:21 -06:00
  • 8302b21f8f Version Bump Jaret Burkett 2026-03-28 13:23:52 -06:00
  • 20929b93df Fix onChange path for EMA Decay input (#695) willhsmit 2026-03-28 12:02:32 -07:00
  • 4ef5cbe5bc Fixing bug where width and height are inverted for control image resizing (#707) abionda-sc 2026-03-28 19:00:32 +00:00
  • 700c4b53d0 Pin timm==1.0.22 (#633) Rob Ballantyne 2026-03-28 18:52:41 +00:00
  • ca72eb1515 Add 1328 native resolution for Qwen Image training (#749) Rayane 2026-03-28 19:09:15 +01:00
  • 5ce87fa48b Version bump Jaret Burkett 2026-03-27 20:26:31 -06:00
  • 740657e25e Improve dataset uploader. Upload the files one at a time instead of one huge chunk. Show progress for each file. Jaret Burkett 2026-03-27 09:26:22 -06:00
  • f85bf065bf Use pooler embeddings for DFE v6 with dino v3 Jaret Burkett 2026-03-27 07:02:07 -06:00
  • a802014ec5 Update the torch versions in the README Jaret Burkett 2026-03-26 12:15:32 -06:00
  • 2782df02c3 Allow HF_HUB_ENABLE_HF_TRANSFER to be set via env variable Jaret Burkett 2026-03-26 10:45:49 -06:00
  • 2c8d2acdcb On jobs table, sort idle jobs by last updated so recent active ones are at the top Jaret Burkett 2026-03-26 10:33:17 -06:00
  • 9a77389653 On a new training job, or when editing one, load everything before allowing editing Jaret Burkett 2026-03-26 10:23:42 -06:00
  • a7bb4ddb2c Work on loss graph. Add smoothed overlay. Allow user to hilite a secton of the graph to zoom into. Jaret Burkett 2026-03-26 10:05:09 -06:00
  • 401f7df425 Merge branch 'main' of github.com:ostris/ai-toolkit Jaret Burkett 2026-03-26 09:11:50 -06:00
  • 4df3b0463f Save job pid to the database and sing sigint to kill it when stopping so it stops immediatly. Jaret Burkett 2026-03-26 09:10:37 -06:00
  • 489b194231 Fix CPU/CUDA device mismatch in Klein edit control image encoding (#742) 科林 KELIN 2026-03-26 01:45:38 +08:00
  • 89d2090962 Fixed race condition that would occasionally set the dataset path to the first one when editing a job Jaret Burkett 2026-03-25 11:22:42 -06:00
  • 3f7a3d8d87 Shorten stal action to 3 months Jaret Burkett 2026-03-25 10:42:18 -06:00
  • 45647c15d3 Added github actions to close stale issues automatically. Hopefully it doesnt break things Jaret Burkett 2026-03-25 10:25:56 -06:00
  • 899ee528f9 Update git ignore Jaret Burkett 2026-03-25 10:16:36 -06:00
  • 5d5a8ef9da Fixed issue with deleting datasets and jobs with newer version of node.js. Bumped minimum version of node js to 20 Jaret Burkett 2026-03-25 10:04:36 -06:00
  • dfde30f231 Fix issue with ltx2 custom te repo path Jaret Burkett 2026-03-25 09:50:18 -06:00
  • b8000dbcbc Bump version Jaret Burkett 2026-03-25 08:18:42 -06:00
  • 54f4732c9b Fix the bug in temporal_compression data loader (#754) Rodrigo Reis 2026-03-25 11:16:44 -03:00
  • 7f3309b291 Add support for audo frame count so datasets can have varrying length videos. Varous ltx 2.3 VAE optimizations such as removing tiling articacts, and doing frame split encoding to reduce vram on encoding/decoding. Jaret Burkett 2026-03-24 12:20:09 -06:00
  • 4ad14d211a Add an import config button (#733) Rasmus Lerdorf 2026-03-23 17:41:27 -04:00
  • 7a0bbca5b1 Fix random_noise_multiplier (#738) Remix 2026-03-23 21:22:16 +00:00
  • 99a4a5887b Fix Qwen attention mask crash with diffusers >=0.37 (#748) Rayane 2026-03-23 21:43:08 +01:00
  • 295094b4b5 Fixed new breaking change in diffusers with with qwen image Jaret Burkett 2026-03-23 14:10:55 -06:00
  • 5642b656b9 Fix audio issues with ltx2 models. Silent codec fails now raised. Auto convert surround sound audio to stereo. Invalidate old caches just to be safe so they recache now. Jaret Burkett 2026-03-23 20:08:33 +00:00
  • 561e6f201c Fixed an issue with ltx 2.3 i2v training Jaret Burkett 2026-03-23 12:41:18 -06:00