Skip to content

Commit 88be559

Browse files
committed
Update tasks.py
1 parent 13391f4 commit 88be559

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

scripts/tasks.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,9 +115,12 @@
115115
'distilroberta-base',
116116
'roberta-large-mnli',
117117

118-
# TODO:
119-
# https://github.com/huggingface/optimum/issues/1028
120-
# 'sentence-transformers/all-distilroberta-v1',
118+
'sentence-transformers/all-distilroberta-v1',
119+
},
120+
'sam': {
121+
'facebook/sam-vit-base',
122+
'facebook/sam-vit-large',
123+
'facebook/sam-vit-huge',
121124
},
122125
'squeezebert': {
123126
'squeezebert/squeezebert-uncased',

0 commit comments

Comments
 (0)