Skip to content

Commit eb38736

Browse files
committed
Fix type annotation of _choose_qparams_affine/gguf(): input = None
1 parent a200ad5 commit eb38736

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchao/quantization/quant_primitives.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1514,7 +1514,7 @@ def choose_qparams_affine_with_min_max(
15141514

15151515
@register_custom_op
15161516
def _choose_qparams_affine(
1517-
input: Optional[torch.Tensor],
1517+
input: torch.Tensor,
15181518
mapping_type: str,
15191519
block_size: List[int],
15201520
target_dtype: torch.dtype,
@@ -1667,7 +1667,7 @@ def reshape_w(w):
16671667

16681668

16691669
def _choose_qparams_gguf(
1670-
input: Optional[torch.Tensor],
1670+
input: torch.Tensor,
16711671
block_size: List[int],
16721672
target_dtype: torch.dtype,
16731673
) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor]:

0 commit comments

Comments
 (0)