kohya-ss / kohya-ss/sd-scripts

TF32 support request.

Open
#245 3 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
7.2k
Forks
1.2k
Avg merge
11m
Merged PRs (30d)
2

Description

https://developer.nvidia.com/blog/accelerating-ai-training-with-tf32-tensor-cores/
https://huggingface.co/docs/transformers/v4.15.0/performance#performance-and-scalability-how-to-fit-a-bigger-model-and-train-it-faster

TF32 is a better bf16 and it would be nice to train with a higher precision.

It seems to be [trivial ](https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_lora.py#L501)to add but I failed trying to add it. Because I saw no differences.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reading the linked NVIDIA and Transformers references, then inspect the referenced examples/text_to_image/train_text_to_image_lora.py location and compare its behavior with this project's training options. Establish a reproducible way to verify whether TF32 is enabled and affects training; done should include a demonstrated result or a documented explanation of why no difference appears.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.