ModelUpdateParams#

class nemo_microservices.types.ModelUpdateParams

Bases: TypedDict

namespace: Required[str]
api_endpoint: APIEndpointData

Data about an API endpoint.

artifact: ModelArtifact

Data about a model artifact (a set of checkpoint files, configs, and other auxiliary info).

The files_url field can point to a DataStore location.

Example:

  • nds://models/rdinu/my-lora-customization

The rdinu/my-lora-customization part above is the actual repository.

If a specific revision needs to be referred, the HuggingFace syntax is used.

  • nds://models/rdinu/my-lora-customization@v1

  • nds://models/rdinu/my-lora-customization@8df79a8

base_model: str | ModelParam

Link to another model which is used as a base for the current model.

custom_fields: Dict[str, str]

A set of custom fields that the user can define and use for various purposes.

description: str

The description of the entity.

guardrails: GuardrailConfigParam

A guardrail configuration

ownership: Ownership

Information about ownership of an entity.

If the entity is a namespace, the access_policies will typically apply to all entities inside the namespace.

peft: ParameterEfficientFinetuningData

Data about a parameter-efficient finetuning.

project: str

The URN of the project associated with this entity.

prompt: PromptData

Prompt engineering data.

schema_version: str

The version of the schema for the object. Internal use only.

spec: ModelSpec

Detailed specification about a model.