diff --git a/vall_e/models/experimental.py b/vall_e/models/experimental.py
index 54ffacf..22d7691 100644
--- 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,