We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a418e29 commit ee18e75Copy full SHA for ee18e75
fms_mo/quant/quantizers.py
@@ -4586,7 +4586,7 @@ def transformers_prepare_input(
4586
if isinstance(data, Mapping):
4587
return type(data)(
4588
{k: transformers_prepare_input(v, dev=dev) for k, v in data.items()}
4589
- )
+ ) # type: ignore[call-arg]
4590
if isinstance(data, (tuple, list)):
4591
return type(data)(transformers_prepare_input(v, dev=dev) for v in data)
4592
if isinstance(data, torch.Tensor):
0 commit comments