tweak imports

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, SamModel, NomicBertModel
+from transformers import BitsAndBytesConfig, AutoConfig, AutoModelForCausalLM, LlavaNextForConditionalGeneration, LlavaForConditionalGeneration, PaliGemmaForConditionalGeneration, Idefics2ForConditionalGeneration, Owlv2ForObjectDetection, GroundingDinoForObjectDetection, SamModel
 import spaces
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
@@ -22,7 +22,6 @@ ARCHITECTURE_MAP = {
     "GroundingDinoForObjectDetection": GroundingDinoForObjectDetection,
     "SamModel": SamModel,
     "AutoModelForCausalLM": AutoModelForCausalLM,
-    "NomicBertModel": NomicBertModel,
 }