Spaces:
Running
on
Zero
Running
on
Zero
Update model_utils.py
Browse filesadd Owlv2ForObjectDetection
- 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 |
|