Merge pull request #1208 from bghira/chore/apple-quanto-dependency

apple: update quanto dependency to upstream repository
bghira · Dec 11, 2024 · d3cb505 · d3cb505
2 parents ea66ac8 + 71bd7e6
commit d3cb505
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/helpers/training/trainer.py b/helpers/training/trainer.py
@@ -2987,7 +2987,7 @@ def train(self):
                 }
                 if self.grad_norm is not None:
                     if self.config.grad_clip_method == "norm":
-                        logs["grad_norm"] = self.grad_norm
+                        logs["grad_norm"] = float(self.grad_norm.clone().detach())
                     elif self.config.grad_clip_method == "value":
                         logs["grad_absmax"] = self.grad_norm
 

diff --git a/install/apple/poetry.lock b/install/apple/poetry.lock