tonera
/

Chroma1-HD-SVDQ

@@ -52,8 +52,12 @@ pip install https://github.com/nunchaku-ai/nunchaku/releases/download/vX.Y.Z/nun
 - **Tip (RTX 50 series GPUs)**: usually `CUDA >= 12.8` is recommended, and FP4 models are preferred for better compatibility and performance (follow the official docs).
 ## Usage example (Diffusers + Nunchaku Transformer)
-Note: I am pushing for the official Nunchaku PR to be merged: https://github.com/nunchaku-ai/nunchaku/pull/916
-Until then, if you want to try it out, you can copy `transformer_chroma.py` from the repository to `nunchaku/models/transformers/transformer_chroma.py`.This is a relatively low-performance version; a high-performance version still needs to wait for the PR to be approved.
 ```python
 import torch

 - **Tip (RTX 50 series GPUs)**: usually `CUDA >= 12.8` is recommended, and FP4 models are preferred for better compatibility and performance (follow the official docs).
 ## Usage example (Diffusers + Nunchaku Transformer)
+Note: I am pushing for the official Nunchaku PR to be merged: https://github.com/nunchaku-ai/nunchaku/pull/928
+Until then, if you want to try it out, you can copy `transformer_chroma.py` from the repository to `nunchaku/models/transformers/transformer_chroma.py`.
+Usage like this:
+```
+from nunchaku.models.transformers.transformer_chroma import NunchakuChromaTransformer2dModel
+```
 ```python
 import torch