Skip to content

Gradio UI for training video models using finetrainers

License

Notifications You must be signed in to change notification settings

neph1/finetrainers-ui

Repository files navigation

finetrainers-ui

Simple GUI for finetrainers (but in no way affiliated with it) made using gradio, inspired by bmalthais gui for kohya-ss scripts.

The goal is to be able to reproduce and analyze previous runs by saving meta data.

Doesn't contain finetrainers. You must download it and get it running separately.

I try to keep main stable, but if it fails, step back one version and try that.

Screenshot from 2024-12-30 07-56-37

Recent changes:

v0.12.0 Compatible with finetrainers v0.1.0 (Prelease)

v0.11.0 Settings for Wan (finetrainers v0.0.1)

v0.10.0: fp8 training

I'm migrating to the new finetrainers structure. Currently tested with LTX-Video. Keeping legacy CogVideoX trainer in the "Legacy" tab.

Usage:

Have CogvideoX-Factory finetrainers

Clone repo

Install streamlit pip install gradio

Run with python app.py

Config path to cogvideox-factory in app

Configure your runtime settings

Press run

Cross fingers and hope it works

Early days: May not work as expected.

The editor is pretty dumb at the moment, and formatting is expected to be the same as in the .sh files, ie spaces between buckets, comma between gpu_ids.