IntelLabs
/

LlavaOLMoBitnet1B

Model card Files Files and versions

naveensp commited on Aug 22, 2024

Commit

0cc08b4

·

verified ·

1 Parent(s): 12f1a9e

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -59,7 +59,7 @@
   "use_mm_proj": true,
   "auto_map": {
     "AutoConfig": "llava.model.language_model.llava_olmo1p58b.LlavaOLMoBitnet1BConfig",
-    "LlavaForConditionalGeneration": "llava.model.language_model.llava_olmo1p58b.LlavaOLMoBitnet1BForCausalLM"
   },
   "image_aspect_ratio": "pad"

   "use_mm_proj": true,
   "auto_map": {
     "AutoConfig": "llava.model.language_model.llava_olmo1p58b.LlavaOLMoBitnet1BConfig",
+    "AutoModelForCausalLM": "llava.model.language_model.llava_olmo1p58b.LlavaOLMoBitnet1BForCausalLM"
   },
   "image_aspect_ratio": "pad"