dnrops
/
transformers.js
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update supported_models.py

This commit is contained in:
Joshua Lochner 2023-05-31 15:07:54 +02:00
parent 84719a5fa3
commit eed6a05c5f
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/supported_models.py
View File

@ -120,9 +120,8 @@ SUPPORTED_MODELS = {
],
'sam': [
'facebook/sam-vit-base',
# TODO:
# 'facebook/sam-vit-large',
# 'facebook/sam-vit-huge',
'facebook/sam-vit-large',
'facebook/sam-vit-huge',
],
'squeezebert': [
'squeezebert/squeezebert-uncased',
@ -158,6 +157,10 @@ SUPPORTED_MODELS = {
'openai/whisper-base.en',
'openai/whisper-small',
'openai/whisper-small.en',
'openai/whisper-medium',
'openai/whisper-medium.en',
'openai/whisper-large',
'openai/whisper-large-v2',
],
}