Add TVLT#20725
Conversation
|
Thanks a lot for contributing this model! One first remark is that we tried to avoid all-capitals now in model classes to make all model names consistent across the library, so could you rename your classes to Leaving you in the hands of @sanchit-gandhi and @ArthurZucker to finish cleaning the PR and all the tests. |
There was a problem hiding this comment.
Thank you for adding this model 🚀 !
I reviewed (only) the config/model files so far.
Could you first run make style and apply the changes if any modification is produced.
Some of my comments are about style, which could be probably fixed by this comment.
Then go through the comments (which are not fixed by the above comment).
I will check why CI are all red in the meantime.
ArthurZucker
left a comment
There was a problem hiding this comment.
Hey! Very cool model, I like the simplicity 😉
Here are is my first review
|
@NielsRogge @ArthurZucker |
|
Another question is that ProcessorMixin is multimodal processor but initialization requires tokenizer |
|
Hey!
|
We can call them
There's no need for a MultiModalProcessorMixin class I think, as the Processor class is exactly meant for multi-modal models. I think we just need to update processing_utils.py to handle the audio modality as well, cc @amyeroberts |
|
@NielsRogge @zinengtang Yes, a single |
|
@NielsRogge now I addressed the remaining comments. It makes sense to me that TvltForQuestionAnswering is not needed since it is the same as TvltForAudioVisualClassification. |
| import unittest | ||
|
|
||
| import numpy as np | ||
| import torch |
There was a problem hiding this comment.
The current error in the feature extraction tests can be resolved by safely importing torch e.g.
if is_torch_available():
import torch
e.g. here for Speech T5
|
@zinengtang Thanks for adding these final changes! I've left a comment on Once all the tests are green I think we're good to merge! |
|
@amyeroberts Sounds great. Btw there seems to be a fail from other models |
|
@zinengtang It's coming from main, not this branch :) I've just pushed a change on main - #21643 - which should resolve this temporarily for now. Could you rebase to add it to this branch? |
|
@amyeroberts Now it passed the tests! Thanks so much for the help/suggestions all the way. :) |
|
@zinengtang Thanks for all your work on adding this model, it's great to have it added to the repo! Make sure to spread the word about it being available :) |
What does this PR do?
Add TVLT to transformers. This PR implements a original version of the TVLT: Textless Vision-Language Transformer from the original paper from https://arxiv.org/abs/2209.14156
I have provided the model weights here https://huggingface.co/TVLT/tvlt-base
Before submitting
Pull Request section?
Who can review?
Anyone.