Skip to content

chore: merge dev to main - #530

Merged
lmxyy merged 9 commits into
mainfrom
dev
Jul 14, 2025
Merged

chore: merge dev to main#530
lmxyy merged 9 commits into
mainfrom
dev

Conversation

@lmxyy

@lmxyy lmxyy commented Jul 14, 2025

Copy link
Copy Markdown
Collaborator

Motivation

Modifications

Checklist

  • Code is formatted using Pre-Commit hooks.
  • Relevant unit tests are added in the tests directory following the guidance in tests/README.md.
  • README 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.

Bluear7878 and others added 9 commits July 7, 2025 00:26
* fix:txt2_img pipe and controlnet pipe share components

* add the test script

* comfyUI bug fix

* garbage collection

* wrap the test

---------

Co-authored-by: Muyang Li <lmxyy1999@foxmail.com>
…in multi-GPU systems (#404)

* Fix: Correctly specify non-zero GPUs in multi-GPU environments

This commit resolves an issue where the Nunchaku model could not be
correctly initialized and run on a user-specified non-zero GPU in
multi-GPU systems.

Key changes include:
- Using CUDADeviceContext in the FluxModel constructor to ensure
  the model and its submodules are created within the specified GPU context.
- Modifying the logic in FluxModel::forward for copying residual data
  from CPU back to GPU, ensuring it returns to the correct original GPU device.
- Adding explicit CUDA context management in Tensor::copy_ for data
  copy operations involving CUDA devices (H2D, D2H, D2D) to guarantee
  cudaMemcpyAsync executes on the correct device.

These changes allow users to reliably run Nunchaku on any specified
GPU in a multi-GPU setup.

* finish pre-commit
* update sphinx docs

* update the doc configration

* configure doxyfile

* start building the docs

* building docs

* building docs

* update docs

* finish the installation documents

* finish the installation documents

* finish the installation documents

* start using rst

* use rst instead of md

* need to figure out how to maintain rst

* update

* make linter happy

* update

* link management

* rst is hard to handle

* fix the title-only errors

* setup the rst linter

* add the lora files

* lora added, need to be more comprehensive

* update

* update

* finished lora docs

* finished the LoRA docs

* finished the cn docs

* finished the cn docs

* finished the qencoder docs

* finished the cpu offload

* finished the offload docs

* add the attention docs

* finished the attention docs

* finished the fbcache

* update

* finished the pulid docs

* make linter happy

* make linter happy

* add kontext

* update

* add the docs for gradio demos

* add docs for test.py

* add the docs for utils.py

* make the doc better displayed

* update

* update

* add some docs

* style: make linter happy

* add docs

* update

* add caching docs

* make linter happy

* add api docs

* fix the t5 docs

* fix the t5 docs

* fix the t5 docs

* hide the private functions

* update

* fix the docs of caching utils

* update docs

* finished the docstring of nunchaku cahcing

* update packer

* revert the docs

* better docs for packer.py

* better docs for packer.py

* better docs for packer.py

* better docs for packer.py

* update

* update docs

* caching done

* caching done

* lora

* lora

* lora

* update

* python docs

* reorg docs

* add the initial version of faq

* update

* make linter happy

* reorg

* reorg

* add crossref

* make linter happy

* better docs

* make linter happy

* preliminary version of the docs done

* update

* update README

* update README

* docs done

* update README

* update docs

* not using datasets 4 for now
@lmxyy lmxyy changed the title Merge dev to main chore: merge dev to main Jul 14, 2025
@lmxyy
lmxyy merged commit cd21409 into main Jul 14, 2025
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.

3 participants