Skip to content

Commit 85d8b48

Browse files
committed
minor fixes to remove unused headers/lines and add exception
Signed-off-by: genquan9 <[email protected]>
1 parent 88121b9 commit 85d8b48

File tree

3 files changed

+1
-6
lines changed

3 files changed

+1
-6
lines changed

nemo/collections/vlm/gemma3vl/data/task_encoder.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
from nemo.collections.vlm.data.task_encoder import TaskEncoder as BaseTaskEncoder
2626
from nemo.collections.vlm.data.task_encoder import TaskEncoderConfig as BaseTaskEncoderConfig
2727
from nemo.collections.vlm.data.utils import _find_pattern_indices
28-
from nemo.collections.vlm.qwen2vl.data.multimodal_tokens import IGNORE_INDEX, IMAGE_TOKEN_INDEX
2928
from nemo.utils import logging
3029

3130

scripts/vlm/gemma3vl_export.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,6 @@
11
"""Export Gemma3VL NeMo checkpoints to Hugging Face format."""
22

33
import argparse
4-
import importlib
5-
import os
6-
import sys
74
from pathlib import Path
85

96
from 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

scripts/vlm/gemma3vl_finetune.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)