From 2ffad5cb6f24a566ceb40042e2296737d518d183 Mon Sep 17 00:00:00 2001 From: mrq Date: Tue, 4 Jun 2024 14:20:57 -0500 Subject: [PATCH] typo --- vall_e/models/experimental.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,