Skip to content

Commit bebcc55

Browse files
committed
fix typo, comment
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 1e412d8 commit bebcc55

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/compressed_tensors/compressors/model_compressors/model_compressor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def from_compression_config(
147147

148148
sparsity_config = cls.parse_sparsity_config(compression_config)
149149
quantization_config = cls.parse_quantization_config(compression_config)
150-
# NOTE: transfrom config is not support by ctconfig yet
150+
# TODO: transform config is not support by CompressedTensorsConfig yet
151151

152152
if sparsity_config is None and quantization_config is None:
153153
return None

src/compressed_tensors/transform/factory/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,10 @@
1414

1515
from abc import ABC, abstractmethod
1616
from collections import defaultdict
17-
from typing import List, Optional, Tuple, Set
17+
from typing import List, Optional, Set, Tuple
1818

1919
import torch
2020
import torch.nn.utils.parametrize as P
21-
from compressed_tensors.utils.internal import InternalModule
2221
from compressed_tensors.registry.registry import RegistryMixin, T
2322
from compressed_tensors.transform import (
2423
TransformArgs,
@@ -34,6 +33,7 @@
3433
register_offload_module,
3534
update_offload_parameter,
3635
)
36+
from compressed_tensors.utils.internal import InternalModule
3737
from torch import Tensor
3838
from torch.nn import Module, Parameter
3939

0 commit comments

Comments
 (0)