dwb2023 commited on
Commit
5cb1ec3
·
verified ·
1 Parent(s): 7be921e

Update model_utils.py

Browse files

add Owlv2ForObjectDetection

Files changed (1) hide show
  1. model_utils.py +2 -1
model_utils.py CHANGED
@@ -1,7 +1,7 @@
1
  import subprocess
2
  import os
3
  import torch
4
- from transformers import BitsAndBytesConfig, AutoConfig, AutoModelForCausalLM, LlavaNextForConditionalGeneration, LlavaForConditionalGeneration, PaliGemmaForConditionalGeneration, Idefics2ForConditionalGeneration
5
  import spaces
6
 
7
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
@@ -18,6 +18,7 @@ ARCHITECTURE_MAP = {
18
  "LlavaForConditionalGeneration": LlavaForConditionalGeneration,
19
  "PaliGemmaForConditionalGeneration": PaliGemmaForConditionalGeneration,
20
  "Idefics2ForConditionalGeneration": Idefics2ForConditionalGeneration,
 
21
  "AutoModelForCausalLM": AutoModelForCausalLM
22
  }
23
 
 
1
  import subprocess
2
  import os
3
  import torch
4
+ from transformers import BitsAndBytesConfig, AutoConfig, AutoModelForCausalLM, LlavaNextForConditionalGeneration, LlavaForConditionalGeneration, PaliGemmaForConditionalGeneration, Idefics2ForConditionalGeneration, Owlv2ForObjectDetection
5
  import spaces
6
 
7
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 
18
  "LlavaForConditionalGeneration": LlavaForConditionalGeneration,
19
  "PaliGemmaForConditionalGeneration": PaliGemmaForConditionalGeneration,
20
  "Idefics2ForConditionalGeneration": Idefics2ForConditionalGeneration,
21
+ "Owlv2ForObjectDetection": Owlv2ForObjectDetection,
22
  "AutoModelForCausalLM": AutoModelForCausalLM
23
  }
24