Hugging Face's logo

Spaces:
dwb2023
/
hf_extractor
Running on Zero

App Files Community
dwb2023 commited on
Commit
fdfb71c
·
verified ·
1 Parent(s): b770d28

update model architectures

Browse files
Files changed (1) hide show
  1. model_utils.py +2 -1
model_utils.py CHANGED
@@ -21,7 +21,8 @@ ARCHITECTURE_MAP = {
21
  "Owlv2ForObjectDetection": Owlv2ForObjectDetection,
22
  "GroundingDinoForObjectDetection": GroundingDinoForObjectDetection,
23
  "SamModel": SamModel,
24
- "AutoModelForCausalLM": AutoModelForCausalLM
 
25
  }
26
 
27
 
 
21
  "Owlv2ForObjectDetection": Owlv2ForObjectDetection,
22
  "GroundingDinoForObjectDetection": GroundingDinoForObjectDetection,
23
  "SamModel": SamModel,
24
+ "AutoModelForCausalLM": AutoModelForCausalLM,
25
+ "NomicBertModel", NomicBertModel,
26
  }
27
 
28