IntelLabs
/

LlavaOLMoBitnet1B

Model card Files Files and versions

naveensp commited on Aug 22, 2024

Commit

29b160c

·

verified ·

1 Parent(s): ac45a66

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -45,8 +45,8 @@
   "inference_mode":false,
   "weight_tying": true,
   "auto_map": {
-    "AutoConfig": "configuration_olmo.OLMoConfig",
-    "AutoModelForCausalLM": "modeling_olmo.OLMoForCausalLM"
   },
   "freeze_mm_vision_resampler": false,
   "mm_hidden_size": 1024,

   "inference_mode":false,
   "weight_tying": true,
   "auto_map": {
+    "AutoConfig": "llava.model.language_model.llava_olmo1p58b.LlavaOLMoBitnet1BConfig",
+    "AutoModelForCausalLM": "llava.model.language_model.llava_olmo1p58b.LlavaOLMoBitnet1BForCausalLM"
   },
   "freeze_mm_vision_resampler": false,
   "mm_hidden_size": 1024,