Skip to content

Commit 83fcffd

Browse files
committed
type hint nit
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent f324af7 commit 83fcffd

File tree

1 file changed

+1
-1
lines changed
  • src/compressed_tensors/compressors/quantized_compressors

1 file changed

+1
-1
lines changed

src/compressed_tensors/compressors/quantized_compressors/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def decompress(
163163
self,
164164
path_to_model_or_tensors: Union[str, Path, Dict[str, Any]],
165165
names_to_scheme: Dict[str, QuantizationScheme],
166-
device: torch.device = "cpu",
166+
device: torch.device = torch.device("cpu"),
167167
) -> Generator[Tuple[str, Tensor], None, None]:
168168
"""
169169
Reads a compressed state dict located at path_to_model_or_tensors

0 commit comments

Comments
 (0)