Skip to content

Commit ab9f48b

Browse files
fix typehint
1 parent e9f6277 commit ab9f48b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/llmcompressor/transformers/compression/sparsity_metadata_config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from __future__ import annotations
12
from compressed_tensors import CompressionFormat, SparsityCompressionConfig
23
from compressed_tensors.config import SparsityStructure
34
from compressed_tensors.quantization import QuantizationType
@@ -86,7 +87,7 @@ def from_pretrained(
8687
quantization_format: CompressionFormat | None = None,
8788
disable_sparse_compression: bool = False,
8889
sparsity_structure: str | None = None,
89-
) -> "SparsityCompressionConfig" | None:
90+
) -> SparsityCompressionConfig | None:
9091
"""
9192
Determines compression type and informational parameters for a given model
9293

0 commit comments

Comments
 (0)