Update model_utils.py

add Owlv2ForObjectDetection

model_utils.py

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