File tree Expand file tree Collapse file tree 3 files changed +1
-6
lines changed
nemo/collections/vlm/gemma3vl/data Expand file tree Collapse file tree 3 files changed +1
-6
lines changed Original file line number Diff line number Diff line change 2525from nemo .collections .vlm .data .task_encoder import TaskEncoder as BaseTaskEncoder
2626from nemo .collections .vlm .data .task_encoder import TaskEncoderConfig as BaseTaskEncoderConfig
2727from nemo .collections .vlm .data .utils import _find_pattern_indices
28- from nemo .collections .vlm .qwen2vl .data .multimodal_tokens import IGNORE_INDEX , IMAGE_TOKEN_INDEX
2928from nemo .utils import logging
3029
3130
Original file line number Diff line number Diff line change 11"""Export Gemma3VL NeMo checkpoints to Hugging Face format."""
22
33import argparse
4- import importlib
5- import os
6- import sys
74from pathlib import Path
85
96from huggingface_hub import hf_hub_download
@@ -65,7 +62,7 @@ def main():
6562 local_dir = args .output_hf_path ,
6663 )
6764 print (f"Downloaded { downloaded_path } during export gamma3vl models." )
68- except :
65+ except Exception as e :
6966 print (f"Ignore { file_name } during export gamma3vl models." )
7067
7168
Original file line number Diff line number Diff line change @@ -84,7 +84,6 @@ def main(args):
8484 raise ValueError ("data_dir is required for energon data type." )
8585 # Initialize the data module
8686 use_packed_sequence = False
87- hf_processor = Gemma3Processor .from_pretrained (args .hf_model_id )
8887 data = EnergonMultiModalDataModule (
8988 path = args .data_dir ,
9089 tokenizer = tokenizer ,
You can’t perform that action at this time.
0 commit comments