Suggestion for improvement

[chencxt]

With generative AI models being so popular right now, why not add a training log? Even fine-tuning a model with a small number of parameters can take several hours, which is also a good way to pass the time. I suggest adding a training module, preferably with pre-set epochs and processes, and then adding a progress bar for each epoch. That would definitely be interesting.

[svenstaro]

Sounds cool! Would you like to take a stab at this?