Skip to content

Commit e252fb2

Browse files
committed
add image-text data calibration support
Signed-off-by: Zhiyu Cheng <zhiyuc@nvidia.com>
1 parent 52eee84 commit e252fb2

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

examples/llm_ptq/example_utils.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -376,10 +376,7 @@ def get_model(
376376
if is_nemotron_parse:
377377
# Nemotron-Parse works fine with device_map="auto"
378378
# Keep device_map="auto" to ensure proper device placement
379-
print(
380-
"Detected Nemotron-Parse model from config. "
381-
"Using automatic device mapping."
382-
)
379+
print("Detected Nemotron-Parse model from config. Using automatic device mapping.")
383380
else:
384381
# For other Nemotron VL models, disable device_map for compatibility
385382
print(

0 commit comments

Comments
 (0)