Skip to content

Commit 7bce92e

Browse files
committed
add image-text data calibration support
Signed-off-by: Zhiyu Cheng <zhiyuc@nvidia.com>
1 parent 769c04d commit 7bce92e

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
@@ -522,10 +522,7 @@ def get_model(
522522
if is_nemotron_parse:
523523
# Nemotron-Parse works fine with device_map="auto"
524524
# Keep device_map="auto" to ensure proper device placement
525-
print(
526-
"Detected Nemotron-Parse model from config. "
527-
"Using automatic device mapping."
528-
)
525+
print("Detected Nemotron-Parse model from config. Using automatic device mapping.")
529526
else:
530527
# For other Nemotron VL models, disable device_map for compatibility
531528
print(

0 commit comments

Comments
 (0)