Skip to content

Commit b9f17ed

Browse files
committed
fix: ruff import sort order in test_pytorch.py
Signed-off-by: 김대익 <33992354+dik654@users.noreply.github.com>
1 parent 6c0c2c9 commit b9f17ed

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

src/llmcompressor/modifiers/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ bin using a scale and (optional) zero point. This basic quantization algorithm i
3737
suitable for FP8 quantization. A variety of quantization schemes are supported via the
3838
[compressed-tensors](https://github.com/neuralmagic/compressed-tensors) library.
3939

40-
### [GPTQ](./quantization/gptq/base.py)
40+
### [GPTQ](./gptq/base.py)
4141
One-shot algorithm that uses calibration data to select the ideal bin for weight quantization.
4242
This algorithm is applied on top of the basic quantization algorithm, and affects weights only.
4343
The implementation is based on [GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformers](https://arxiv.org/pdf/2210.17323). The algorithm is very similar to SparseGPT: A small amount of calibration data is used
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# Sequential Pipeline #
22
The sequential pipeline is a data pipeline, primarily used for compressing models with the
3-
[GPTQModifier](/src/llmcompressor/modifiers/quantization/gptq/base.py) or the
3+
[GPTQModifier](/src/llmcompressor/modifiers/gptq/base.py) or the
44
[SparseGPTModifier](/src/llmcompressor/modifiers/pruning/sparsegpt/base.py).

tests/llmcompressor/pytorch/modifiers/pruning/sparsegpt/test_pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
import pytest
22
import torch
33

4-
from llmcompressor.modifiers.pruning.sparsegpt import SparseGPTModifier
54
from llmcompressor.modifiers.gptq import GPTQModifier
5+
from llmcompressor.modifiers.pruning.sparsegpt import SparseGPTModifier
66
from tests.llmcompressor.modifiers.conf import LifecyleTestingHarness
77
from tests.llmcompressor.pytorch.helpers import LinearNet
88

0 commit comments

Comments
 (0)