Skip to content

feat: add Turing (20~ Series) GPU compatibility for Z Image Turbo - #833

Merged
lmxyy merged 7 commits into
nunchaku-ai:mainfrom
devgdovg:feat/fp16_compatibility
Jan 11, 2026
Merged

feat: add Turing (20~ Series) GPU compatibility for Z Image Turbo#833
lmxyy merged 7 commits into
nunchaku-ai:mainfrom
devgdovg:feat/fp16_compatibility

Conversation

@devgdovg

@devgdovg devgdovg commented Dec 24, 2025

Copy link
Copy Markdown
Contributor

Motivation

Make Nunchaku Z-Image-Turbo model able to run with Turing GPUs.

Modifications

  • Make SVDQW4A4Linear able to change dtype to torch.float16 when initializing.
  • Manually convert state dict dtype to torch.float16 when loading to Turing GPU in order to avoid range overflow.

Checklist

  • Code is formatted using Pre-Commit hooks.
  • Relevant unit tests are added in the tests directory following the guidance in Contribution Guide.
  • Documentation and example scripts in examples are updated if necessary.
  • Throughput/latency benchmarks and quality evaluations are included where applicable.
  • For reviewers: If you're only helping merge the main branch and haven't contributed code to this PR, please remove yourself as a co-author when merging.
  • Please feel free to join our Slack, Discord or WeChat to discuss your PR.

@anr2me

anr2me commented Dec 29, 2025

Copy link
Copy Markdown

Nice, now we can use it on a free T4 GPUs at Kaggle/Colab 😁

@xynu617

xynu617 commented Dec 31, 2025

Copy link
Copy Markdown

可以在我的Tesla T10上运行了,谢谢

@devgdovg
devgdovg force-pushed the feat/fp16_compatibility branch from b38fa62 to fe31b2d Compare December 31, 2025 17:09
@devgdovg devgdovg changed the title add Turing (20~ Series) GPU compatibility for Z Image Turbo feat: add Turing (20~ Series) GPU compatibility for Z Image Turbo Dec 31, 2025
@ataquino

ataquino commented Jan 8, 2026

Copy link
Copy Markdown

How can I switch to this PR locally to test it out?

@anr2me

anr2me commented Jan 8, 2026

Copy link
Copy Markdown

How can I switch to this PR locally to test it out?

After you git pull the nunchaku repo, do:

git fetch origin pull/833/head:pr-833

and then

git merge pr-833 -m "pr-833 or any message"

or (if it can't be merged successfully)

git checkout pr-833

@lmxyy lmxyy left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

could you add a test for this?

@devgdovg
devgdovg requested a review from lmxyy January 9, 2026 09:01
Comment thread tests/v1/z_image/test_z_image_turbo.py Outdated
@devgdovg
devgdovg requested a review from lmxyy January 9, 2026 09:22

@lmxyy lmxyy left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

passing tests. approved

@lmxyy
lmxyy merged commit 4bc2fbf into nunchaku-ai:main Jan 11, 2026
3 checks passed
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.

5 participants