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

Add blenderbot to supported models

This commit is contained in:
Joshua Lochner 2023-09-11 01:25:48 +02:00
parent fa40795233
commit 0de885a6ac
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
scripts/supported_models.py
View File

@ -97,16 +97,16 @@ SUPPORTED_MODELS = {
'bert-base-chinese',
'emilyalsentzer/Bio_ClinicalBERT',
],
# 'blenderbot': [
# # Text2text generation (TODO add conversational)
# 'facebook/blenderbot-400M-distill',
# 'facebook/blenderbot-1B-distill',
# ],
# 'blenderbot-small': [
# # Text2text generation (TODO add conversational)
# 'facebook/blenderbot-90M', # DEPRECATED
# 'facebook/blenderbot_small-90M',
# ],
'blenderbot': [
# Text2text generation (TODO add conversational)
'facebook/blenderbot-400M-distill',
# 'facebook/blenderbot-1B-distill',
],
'blenderbot-small': [
# Text2text generation (TODO add conversational)
# 'facebook/blenderbot-90M', # DEPRECATED
'facebook/blenderbot_small-90M',
],
'bloom': [
# Text generation
'bigscience/bloom-560m',