kohya-ss / kohya-ss/sd-scripts

how to use flux full-finetune safetensors in diffusers?

Open
#1,626 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

essentially the title. i've tried the following:
```
transformer = diffusers.Flux2DTransformer.from_single_file(path_to_kohya_safetensors, torch.dtype=torch.bfloat16)
pipeline = diffusers.FluxPipeline.from_pretrained(..., transformer=transformer, torch.dtype=torch.bfloat16)
```

however all generated images have weird artifacts, regardless of checkpoint number
i've tried with both lr 1e-5 and 1e-4

Contributor guide

No contributing guide indexed for this repository

Research direction

Start with the shown Flux2DTransformer.from_single_file and FluxPipeline.from_pretrained calls, using the Kohya safetensors checkpoint and the reported dtype settings. Compare the loading path and checkpoint expectations with the generated outputs. Done means identifying a supported usage path or a reproducible cause for the artifacts.

Written by the indexing model from the issue text.

Assessment

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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.