nvidia
/

Llama-3.1-Nemotron-Nano-VL-8B-V1

Image-Text-to-Text

Model card Files Files and versions

amalad commited on Jun 9

Commit

961cb9f

·

1 Parent(s): a02869b

Update model class name

Files changed (2) hide show

config.json +2 -2
modeling.py +1 -1

config.json CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d28e10441f08143e1bad3121dbdb702df97d5ca448758d1982920e05dba0c62d
-size 7662

 version https://git-lfs.github.com/spec/v1
+oid sha256:d249106251318246ac377b0722ad2a733592d54781271e2bcc89ba76626b2edb
+size 7650

modeling.py CHANGED Viewed

@@ -35,7 +35,7 @@ def version_cmp(v1, v2, op='eq'):
     return op_func(version.parse(v1), version.parse(v2))
-class Llama_Nemotron_Nano_VL_Model(PreTrainedModel):
     config_class = Llama_Nemotron_Nano_VL_Config
     main_input_name = 'pixel_values'
     _supports_flash_attn_2 = True

     return op_func(version.parse(v1), version.parse(v2))
+class Llama_Nemotron_Nano_VL(PreTrainedModel):
     config_class = Llama_Nemotron_Nano_VL_Config
     main_input_name = 'pixel_values'
     _supports_flash_attn_2 = True