Skip to content

Commit 32d150c

Browse files
committed
make linter happy
1 parent c92bf55 commit 32d150c

4 files changed

Lines changed: 8 additions & 8 deletions

File tree

docs/source/usage/attention.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,6 @@ and 50-series GPUs compared to FlashAttention-2, without precision loss.
1212

1313
The key change from `Basic Usage <./basic_usage>`_ is use ``transformer.set_attention_impl("nunchaku-fp16")`` to enable FP16 attention.
1414
While FlashAttention-2 is the default, FP16 attention offers better performance on modern NVIDIA GPUs.
15-
Switch back with ``transformer.set_attention_impl("flash-attention2")``.
15+
Switch back with ``transformer.set_attention_impl("flash-attention2")``.
1616

1717
For more details, see :meth:`~nunchaku.models.transformers.transformer_flux.NunchakuFluxTransformer2dModel.set_attention_impl`.

docs/source/usage/lora.rst

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -97,9 +97,9 @@ The syntax is as follows:
9797
This command composes the specified LoRAs with their respective strength values and saves the result to the output file,
9898
which can subsequently be loaded using :meth:`~nunchaku.models.transformers.transformer_flux.NunchakuFluxTransformer2dModel.update_lora_params` for optimized inference performance.
9999

100-
Following composition, users may either load the file directly
101-
(via the ComfyUI LoRA loader or :meth:`~nunchaku.models.transformers.transformer_flux.NunchakuFluxTransformer2dModel.update_lora_params`)
102-
or utilize :mod:`nunchaku.lora.flux.convert` to convert the composed LoRA to Nunchaku's format and export it.
100+
Following composition, users may either load the file directly
101+
(via the ComfyUI LoRA loader or :meth:`~nunchaku.models.transformers.transformer_flux.NunchakuFluxTransformer2dModel.update_lora_params`)
102+
or utilize :mod:`nunchaku.lora.flux.convert` to convert the composed LoRA to Nunchaku's format and export it.
103103
The syntax is as follows:
104104

105105
.. code-block:: bash

docs/source/usage/qencoder.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@ Nunchaku provides a quantized T5 encoder for FLUX.1 to reduce GPU memory usage.
1212
The key changes from `Basic Usage <./basic_usage>`_ are:
1313

1414
**Loading Quantized T5 Encoder** (line 11):
15-
Use :class:`~nunchaku.models.text_encoders.t5_encoder.NunchakuT5EncoderModel` to load the quantized encoder.
15+
Use :class:`~nunchaku.models.text_encoders.t5_encoder.NunchakuT5EncoderModel` to load the quantized encoder.
1616
This reduces GPU memory usage while maintaining quality. Supports local or Hugging Face remote paths.
1717

1818
**Pipeline Integration** (line 14):
19-
Pass the quantized encoder to the pipeline via the ``text_encoder_2`` parameter,
19+
Pass the quantized encoder to the pipeline via the ``text_encoder_2`` parameter,
2020
replacing the default T5 encoder.
2121

2222
.. note::

nunchaku/models/transformers/transformer_flux.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -653,7 +653,7 @@ def set_attention_impl(self, impl: str):
653653
----------
654654
impl : str
655655
Attention implementation to use. Supported values:
656-
656+
657657
- ``"flash-attention2"`` (default): Standard FlashAttention-2.
658658
- ``"nunchaku-fp16"``: Uses FP16 attention accumulation, up to 1.2× faster than FlashAttention-2 on NVIDIA 30-, 40-, and 50-series GPUs.
659659
"""
@@ -783,7 +783,7 @@ def update_lora_params(self, path_or_state_dict: str | dict[str, torch.Tensor]):
783783
----------
784784
path_or_state_dict : str or dict
785785
Path to a LoRA weights file or a state dict. The path supports:
786-
786+
787787
- Local file path, e.g., ``"/path/to/your/lora.safetensors"``
788788
- HuggingFace repo with file, e.g., ``"user/repo/lora.safetensors"``
789789
(automatically downloaded and cached)

0 commit comments

Comments
 (0)