sd-scripts
Add fp8 scaling, fp8 scaled convolutions, and gradient checkpointing for Flux AutoEncoder
#2231
Open
Open
Add fp8 scaling, fp8 scaled convolutions, and gradient checkpointing for Flux AutoEncoder #2231
rockerBOO
wants to merge
6 commits into
kohya-ss:feat-flux-chroma-fp8-scaled
from
rockerBOO:flux-fp8-scaled-ae
Add fp8 scaling for Flux AutoEncoder, and convolutions
f5ed1081
Revert library/custom_train_functions.py
56b949dd
Revert train_network.py
c5a0231e
Fix fp8_scaled_ae parameter
625896c4
Cleanup
d789c6de
Add quantize_conv_weight test
1cd13d9a
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub