nvidia
/

DAM-3B-Self-Contained

Image-Text-to-Text

Model card Files Files and versions Community

richardaecn commited on about 17 hours ago

Commit

19415aa

·

verified ·

1 Parent(s): 90e0439

Upload llava_llama.py

Files changed (1) hide show

llava_llama.py +1 -1

llava_llama.py CHANGED Viewed

@@ -1209,7 +1209,7 @@ class LlavaMetaModel(ABC):
         # print(llm_cfg, vision_tower_cfg, mm_projector_cfg); input("DEBUG load_pretrained")
         with ContextManagers(
             [
-                no_init_weights(_enable=True),
             ]
         ):
             vlm = cls(config, *args, **kwargs)

         # print(llm_cfg, vision_tower_cfg, mm_projector_cfg); input("DEBUG load_pretrained")
         with ContextManagers(
             [
+                no_init_weights(),
             ]
         ):
             vlm = cls(config, *args, **kwargs)