typo

2024-06-04 14:20:57 -05:00 · 2024-06-04 14:20:57 -05:00 · 2ffad5cb6f
commit 2ffad5cb6f
parent 406ff7bbe1
1 changed files with 2 additions and 2 deletions
--- a/vall_e/models/experimental.py
+++ b/vall_e/models/experimental.py
@ -172,8 +172,8 @@ class Model(LlmArchClass):
 			if "do_sample" in kwargs:
 				kwargs.pop("do_sample")

-		return super().forward(*args, **kwargs)
-		
+		return super().generate(*args, **kwargs)
+
 	def forward(
 		self,
 		*args,