nvidia
/

NVIDIA-Nemotron-Parse-v1.1

Image-Text-to-Text

feature-extraction

Model card Files Files and versions

katerynaCh commited on 20 days ago

Commit

3c63eb4

·

verified ·

1 Parent(s): 04e940a

Upload folder using huggingface_hub

Files changed (1) hide show

hf_nemotron_parse_modeling.py +1 -1

hf_nemotron_parse_modeling.py CHANGED Viewed

@@ -347,7 +347,7 @@ class NemotronParsePreTrainedModel(PreTrainedModel):
     An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
     """
     config_class = NemotronParseConfig
-    base_model_prefix = "nemotron_parse" #vision_encoder_decoder"  # Use VisionEncoderDecoder prefix
     main_input_name = "pixel_values"
     supports_gradient_checkpointing = True
     _no_split_modules = ["RadioWithNeck", "MBartDecoder"]

     An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
     """
     config_class = NemotronParseConfig
+    base_model_prefix = "vision_encoder_decoder"  # Use VisionEncoderDecoder prefix
     main_input_name = "pixel_values"
     supports_gradient_checkpointing = True
     _no_split_modules = ["RadioWithNeck", "MBartDecoder"]