Use eval() instead of train(False)

Darwinkel · Apr 2, 2024 · 7dcb6b2 · 7dcb6b2
1 parent 8bfb719
commit 7dcb6b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shepardtts/utils.py b/shepardtts/utils.py
@@ -28,11 +28,11 @@ def load_checkpoint() -> ShepardXtts:
         use_deepspeed=use_deepspeed,
     )
     model.to(settings.DEVICE)
+    model.eval()
 
     if settings.DEVICE == "cpu":
         import intel_extension_for_pytorch as ipex
 
-        model.train(mode=False)
         model = ipex.optimize(model, weights_prepack=False, dtype=torch.float16)
         model = torch.compile(model, backend="ipex")