From a80820dd2358a3bddfc0d93ab23af7ee94ed1ca1 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 30 Oct 2024 02:51:02 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- auto_round/autoround.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/auto_round/autoround.py b/auto_round/autoround.py index 48d7e518..80c145d0 100644 --- a/auto_round/autoround.py +++ b/auto_round/autoround.py @@ -684,7 +684,7 @@ def forward(m, hidden_states=None, *positional_args, **kwargs): for key in kwargs.keys(): if isinstance(kwargs[key], torch.Tensor) or isinstance(kwargs[key], list) \ or isinstance(kwargs[key], tuple): - if key not in self.inputs[name].keys(): # initilization + if key not in self.inputs[name].keys(): # initialization data = to_device(kwargs[key], device=torch.device("cpu")) if data is None or (self.train_bs > 1 and key in shareable_keywords): self.inputs[name][key] = data