ModelConfig

class baseten.client.modelconfig.ModelConfig(*, model_name=None, model_metadata=None, description=None, examples_filename='examples.yaml', data_dir='data', external_data=None, external_package_dirs=None, python_version='py313', base_image=None, requirements_file=None, requirements=None, system_packages=None, environment_variables=None, secrets=None, resources=None, runtime=None, build=None, build_commands=None, docker_server=None, model_cache=None, weights=None, trt_llm=None, training_checkpoints=None, bis_llm=None, input_type='Any', model_framework='custom', model_type='Model', model_module_dir='model', model_class_filename='model.py', model_class_name='Model', bundled_packages_dir='packages', use_local_src=False, cache_internal=None, live_reload=False, apply_library_patches=True, spec_version='2.0', **extra_data)

Bases: BaseModel

Parameters:

• model_name (str | None)

• model_metadata (ModelMetadata | None)

• description (str | None)

• examples_filename (str | None)

• data_dir (str | None)

• external_data (ExternalData | None)

• external_package_dirs (list[str] | None)

• python_version (str | None)

• base_image (BaseImage | None)

• requirements_file (str | None)

• requirements (list[str] | None)

• system_packages (list[str] | None)

• environment_variables (dict[str, str] | None)

• secrets (dict[str, str | None] | None)

• resources (Resources | None)

• runtime (Runtime | None)

• build (Build | None)

• build_commands (list[str] | None)

• docker_server (DockerServer | None)

• model_cache (ModelCache | None)

• weights (Weights | None)

• trt_llm (TRTLLMConfiguration | None)

• training_checkpoints (CheckpointList | None)

• bis_llm (BISLLM | None)

• input_type (str | None)

• model_framework (str | None)

• model_type (str | None)

• model_module_dir (str | None)

• model_class_filename (str | None)

• model_class_name (str | None)

• bundled_packages_dir (str | None)

• use_local_src (bool | None)

• cache_internal (CacheInternal | None)

• live_reload (bool | None)

• apply_library_patches (bool | None)

• spec_version (str | None)

• extra_data (Any)

model_config = {'extra': 'allow'}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].