Update model_utils.py

model_utils.py

@@ -1,7 +1,7 @@
 import subprocess
 import os
 import torch
-from transformers import BitsAndBytesConfig, AutoConfig, AutoModelForCausalLM, LlavaNextForConditionalGeneration, LlavaForConditionalGeneration, PaliGemmaForConditionalGeneration, Idefics2ForConditionalGeneration, Owlv2ForObjectDetection, GroundingDinoForObjectDetection
+from transformers import BitsAndBytesConfig, AutoConfig, AutoModelForCausalLM, LlavaNextForConditionalGeneration, LlavaForConditionalGeneration, PaliGemmaForConditionalGeneration, Idefics2ForConditionalGeneration, Owlv2ForObjectDetection, GroundingDinoForObjectDetection, SamModel
 import spaces
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
@@ -20,6 +20,7 @@ ARCHITECTURE_MAP = {
     "Idefics2ForConditionalGeneration": Idefics2ForConditionalGeneration,
     "Owlv2ForObjectDetection": Owlv2ForObjectDetection,
     "GroundingDinoForObjectDetection": GroundingDinoForObjectDetection,
+    "SamModel": SamModel,
     "AutoModelForCausalLM": AutoModelForCausalLM
 }