Skip to content

Flux2 Updates to Improve Image Quality#120

Merged
kevchengcodes merged 1 commit into
apple:mainfrom
kevchengcodes:flux2_fixes
Jul 21, 2026
Merged

Flux2 Updates to Improve Image Quality#120
kevchengcodes merged 1 commit into
apple:mainfrom
kevchengcodes:flux2_fixes

Conversation

@kevchengcodes

Copy link
Copy Markdown
Contributor

Summary

Issue #111 led us to investigate what might be causing Flux2 regressions. This PR makes a few small changes in the pipeline to be more faithful to the original Python definitions. Although these lead to meaningfully better results, we may need to conduct further investigation to resolve any other possible issues. Therefore, follow-up PRs may be made to further improve image quality and determine root causes.

Changes

  1. Denoising noise schedule computation (mu and sigma)
  2. Disable thinking for text encoder
  3. Use <|endoftext|> token (151643) instead of <|im_end|> (151645) for padding

Results

Using similar prompts to those described in #111, we see images like below (a prompt each for the lobster, butterfly, ant, and bee)
image

@kevchengcodes
kevchengcodes requested a review from stikves July 21, 2026 19:32
@kevchengcodes kevchengcodes self-assigned this Jul 21, 2026
@kevchengcodes
kevchengcodes merged commit ca4fa50 into apple:main Jul 21, 2026
3 checks passed
@kevchengcodes
kevchengcodes deleted the flux2_fixes branch July 21, 2026 21:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants