diff --git a/auto_gptq/modeling/_base.py b/auto_gptq/modeling/_base.py index 89b5ae7..3423be5 100644 --- a/auto_gptq/modeling/_base.py +++ b/auto_gptq/modeling/_base.py @@ -433,7 +433,8 @@ class BaseGPTQForCausalLM(nn.Module, PushToHubMixin): return torch.device(device) def to(self, device: Union[str, torch.device]): - return self.model.to(device) + self.model.to(device) + return self def forward(self, *args, **kwargs): return self.model(*args, **kwargs)