Rm typing

huggingface · Oct 22, 2024 · 0757cbf · 0757cbf
1 parent 4663746
commit 0757cbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/accelerate/optimizer.py b/src/accelerate/optimizer.py
@@ -197,7 +197,7 @@ def __setstate__(self, state):
  self._optimizer_original_step_method = self.optimizer.step
  self._optimizer_patched_step_method = patch_optimizer_step(self, self.optimizer.step)
 
- def multiply_grads(self, constant: float | torch.Tensor) -> None:
+ def multiply_grads(self, constant) -> None:
  """
  Multiplies the gradients of the parameters by a constant. Needed during gradient accumulation.