AfterHAL

Hi. I use to forget to close TagGui after generating Auto-Captioning and the VRAM is not free for another usage. It would be nice to have an "Unload Model after...

Can't install. Error

12

Hi. I've been trying to install torch-cubas-hgemm (for flux-fp8-api) on ubuntu without success. The same error `ModuleNotFoundError: No module named 'torch'`, whether with `pip install git+https://github.com/aredden/torch-cublas-hgemm.git@master` or with `python -m...

Problem launching training with less GPUs

10

Hi. I setup a Sana training session with one 4090 GPU on a PC, everything was fine so I moved the config and the checkpoint to a PC with 7...

Wrong epoch count on training (Bug?)

1

Hi. I didn't pay attention to it before, but I am currently testing a training base on the "Sana_600M_img512.yaml" config file with a dataset ok 20k images (single 4090 GPU...

Flux LoRA training relaunch error when using Automagic Optimizer.

2

## This is for bugs only Did you already ask [in the discord](https://discord.gg/VXmU2f5WEU)? No You verified that this is a bug and not a feature request or question by asking...

I did a fresh install od Ubuntu and ai-toolkit today and got this error running the script : ``` RuntimeError: cuDNN Frontend error: [cudnn_frontend] Error: No execution plans support the...