Update model_utils.py

add GroundingDinoForObjectDetection

model_utils.py

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