Commit 063ef88d authored by wenjh's avatar wenjh
Browse files

Merge nv main up to v2.10.0.dev0


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parents 91670b05 5624dbb4
# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# #
# See LICENSE for license information. # See LICENSE for license information.
import pathlib, os import pathlib
from nvdlfw_inspect.config_manager import ConfigManager from nvdlfw_inspect.config_manager import ConfigManager
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
...@@ -110,9 +110,9 @@ def _train(args): ...@@ -110,9 +110,9 @@ def _train(args):
build_model_context = nullcontext build_model_context = nullcontext
build_model_context_args = {} build_model_context_args = {}
from transformer_engine.pytorch import fp8_model_init from transformer_engine.pytorch import quantized_model_init
build_model_context = fp8_model_init build_model_context = quantized_model_init
build_model_context_args["enabled"] = True build_model_context_args["enabled"] = True
# Build the model with the specified context # Build the model with the specified context
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment