set model to eval when inferencing (very important)

.gitignore

@@ -7,3 +7,4 @@ __pycache__
 /*.egg-info
 /vall_e/version.py
 /build
+/.cache

vall_e/inference.py

@@ -24,6 +24,9 @@ class TTS():
 		else:
 			self.load_models( config )
 
+		self.ar.eval()
+		self.nar.eval()
+
 		self.loading = False 
 
 	def load_models( self, config_path ):

vall_e/train.py

@@ -5,7 +5,7 @@ from .data import create_train_val_dataloader
 from .emb import qnt
 
 from .utils import setup_logging, to_device, trainer, flatten_dict, do_gc
-
+from .utils.trainer import load_engines
 
 import auraloss
 import json