MCPcopy
hub / github.com/NVIDIA/TensorRT-LLM / save_config

Method save_config

tensorrt_llm/builder.py:447–450  ·  view source on GitHub ↗
(builder_config: BuilderConfig, config_path: str)

Source from the content-addressed store, hash-verified

445
446 @staticmethod
447 def save_config(builder_config: BuilderConfig, config_path: str):
448 config = builder_config.to_dict()
449 to_json_file(config, config_path)
450 logger.info(f'Config saved to {config_path}.')
451
452
453class BuildConfig(BaseModel):

Callers 7

build_bertFunction · 0.95
test_basic_builder_flowMethod · 0.95
testFunction · 0.95
build_engineFunction · 0.80
build_engineFunction · 0.80
build_sdxl_unet.pyFile · 0.80
build_trt_engineFunction · 0.80

Calls 3

to_json_fileFunction · 0.85
to_dictMethod · 0.45
infoMethod · 0.45

Tested by 1

test_basic_builder_flowMethod · 0.76