Spaces:

dwb2023
/

hf_extractor

Running on Zero

dwb2023 commited on 12 days ago

Commit

71559ee

•

1 Parent(s): 7ed52f5

tweak imports

Files changed (1) hide show

model_utils.py CHANGED Viewed

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

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