From db181f8e88bb150b6e0d01cdd486a7f57f350248 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Mon, 24 Feb 2025 21:07:44 -0600
Subject: [PATCH] only do auto=equal for nemo as its an FSQ

---
 vall_e/models/ar_nar.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/vall_e/models/ar_nar.py b/vall_e/models/ar_nar.py
index eb36b7a..7fa6f3e 100644
--- 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