From f593ee98fcf0cd110d073e11d2e9f13e0d5a9119 Mon Sep 17 00:00:00 2001 From: mrq Date: Sun, 23 Feb 2025 21:20:36 -0600 Subject: [PATCH] ugh --- vall_e/engines/__init__.py | 2 +- vall_e/engines/base.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/vall_e/engines/__init__.py b/vall_e/engines/__init__.py index 65672cc..d745864 100755 --- a/vall_e/engines/__init__.py +++ b/vall_e/engines/__init__.py @@ -60,7 +60,7 @@ def load_engines(training=True, **model_kwargs): # to handle the issue of training with deepspeed, but inferencing with local if checkpoint_path.exists() and backend == "local": - tag = open(checkpoint_path).read() + tag = open(checkpoint_path).read().strip() checkpoint_path = pick_path( checkpoint_path.parent / tag / f"state.{cfg.weights_format}", *[ f'.{format}' for format in cfg.supported_weights_formats] ) # if loaded using --model= diff --git a/vall_e/engines/base.py b/vall_e/engines/base.py index f8ece81..8ffc733 100755 --- a/vall_e/engines/base.py +++ b/vall_e/engines/base.py @@ -258,7 +258,7 @@ class Engine(): self.optimizer.step() self.optimizer.zero_grad() - self._get_grad_norm() + # self._get_grad_norm() # doesn't actually work def _get_grad_norm(self):