SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add Blip and Blip2 for pipeline tests (#21904)

Browse Source 
* fix

* add to tests

* style and quality

* add missing

---------

Co-authored-by: NielsRogge <NielsRogge@users.noreply.github.com>
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-03-02 18:20:34 +01:00
committed by GitHub
parent 1325459105
commit e6de918676
5 changed files with 66 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
tests/utils/tiny_model_summary.json
View File

@@ -55,6 +55,42 @@
],
"processor_classes": []
},
"BlipModel": {
"tokenizer_classes": [
"BertTokenizerFast",
"BertTokenizer"
],
"processor_classes": [
"BlipImageProcessor"
]
},
"BlipForConditionalGeneration": {
"tokenizer_classes": [
"BertTokenizerFast",
"BertTokenizer"
],
"processor_classes": [
"BlipImageProcessor"
]
},
"Blip2Model": {
"tokenizer_classes": [
"GPT2TokenizerFast",
"GPT2Tokenizer"
],
"processor_classes": [
"BlipImageProcessor"
]
},
"Blip2ForConditionalGeneration": {
"tokenizer_classes": [
"GPT2TokenizerFast",
"GPT2Tokenizer"
],
"processor_classes": [
"BlipImageProcessor"
]
},
"BloomModel": {
"tokenizer_classes": [
"BloomTokenizerFast"