From eed6a05c5f172f0c1b7477df8476fadf67e880bc Mon Sep 17 00:00:00 2001
From: Joshua Lochner <admin@xenova.com>
Date: Wed, 31 May 2023 15:07:54 +0200
Subject: [PATCH] Update supported_models.py

---
 scripts/supported_models.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/scripts/supported_models.py b/scripts/supported_models.py
index bce8161..ae8f6ba 100644
--- a/scripts/supported_models.py
+++ b/scripts/supported_models.py
@@ -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',
     ],
 }