mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

typo

Browse Source
This commit is contained in:
mrq 2024-06-04 14:20:57 -05:00
parent 406ff7bbe1
commit 2ffad5cb6f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vall_e/models/experimental.py
View File

@ -172,8 +172,8 @@ class Model(LlmArchClass):
if "do_sample" in kwargs: if "do_sample" in kwargs:
kwargs.pop("do_sample") kwargs.pop("do_sample")
return super().forward(*args, **kwargs) return super().generate(*args, **kwargs)
def forward( def forward(
self, self,
*args, *args,