-
-
Notifications
You must be signed in to change notification settings - Fork 315
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
update modules #59
update modules #59
Conversation
valle/models/valle.py
Outdated
self.predict_layers[j].weight = self.nar_embeddings[j + 1].weight | ||
for j in range(0, 6): | ||
self.nar_predict_layers[j].weight = self.nar_audio_embeddings[ | ||
j + 2 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里的embedding share为什么只有6层,而且是j+2,我看论文里好像是说,predict_layer只有7层,embedding_layer是8层,所以predict的第1层和embedding的第2层共享,是这样嘛
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Good Work
demo page https://lifeiteng.github.io/valle/index.html one GPU with 24GB memory
|
This maybe the last breaking modification!