Training added on top of flux_impl#147

Merged

entrpn merged 9 commits intoAI-Hypercomputer:mainfrom

ksikiric:kris/flux-impl-training

Apr 16, 2025

Commits on Apr 16, 2025

Added training code, loss and results are stable
ksikiric
committed
Rebased on flux_lora and aligned flux_pipeline with changes in generate_flux.py
ksikiric
committed
batch text encoding.

jfacevedo-google
authored and
ksikiric
committed
comment out post_training_steps

jfacevedo-google
authored and
ksikiric
committed
refactor some code for similarity to sd trainers.

jfacevedo-google
authored and
ksikiric
committed
Added orbax saving and a new file for inference that utilizes the pipeline.
ksikiric
committed
Update generate_flux_pipeline.py

entrpn
authored and
ksikiric
committed
Fixed comments and rebased on main
ksikiric
committed
ruff + code_style
ksikiric
committed