only do auto=equal for nemo as its an FSQ

2025-02-24 21:07:44 -06:00 · 2025-02-24 21:07:44 -06:00 · db181f8e88
commit db181f8e88
parent a5a04c39ef
1 changed files with 2 additions and 0 deletions
--- a/vall_e/models/ar_nar.py
+++ b/vall_e/models/ar_nar.py
@ -78,6 +78,8 @@ class AR_NAR(Base):

 		if self.version >= 7:
 			masking_train_rvq_levels = [0,self.n_resp_levels]
+			
+		if cfg.audio_backend == "nemo":
 			rvq_levels_p = [ i for i in range( quant_level_range[0], quant_level_range[1] + 1 ) ]

 		# CFG