nemo_microservices.types.model_search

Module Contents

Classes

ModelSearch

Data

APIEndpoint
Artifact
Guardrails
Ownership
Peft
Prompt
Spec

API

nemo_microservices.types.model_search.APIEndpoint: typing_extensions.TypeAlias

None

nemo_microservices.types.model_search.Artifact: typing_extensions.TypeAlias

None

nemo_microservices.types.model_search.Guardrails: typing_extensions.TypeAlias

None

class nemo_microservices.types.model_search.ModelSearch(/, **data: Any)

Bases: nemo_microservices._models.BaseModel

api_endpoint: nemo_microservices.types.model_search.APIEndpoint | None

None

Data about an API endpoint.

artifact: nemo_microservices.types.model_search.Artifact | None

None

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 | List[str] | None

None

created_at: nemo_microservices.types.shared.date_range.DateRange | None

None

custom_fields: Dict[str, object] | List[Dict[str, object]] | None

None

description: str | List[str] | None

None

guardrails: nemo_microservices.types.model_search.Guardrails | None

None

A guardrail configuration

name: str | List[str] | None

None

namespace: str | List[str] | None

None

ownership: nemo_microservices.types.model_search.Ownership | None

None

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: nemo_microservices.types.model_search.Peft | None

None

Data about a parameter-efficient finetuning.

project: str | List[str] | None

None

prompt: nemo_microservices.types.model_search.Prompt | None

None

Prompt engineering data.

spec: nemo_microservices.types.model_search.Spec | None

None

Detailed specification about a model.

updated_at: nemo_microservices.types.shared.date_range.DateRange | None

None

nemo_microservices.types.model_search.Ownership: typing_extensions.TypeAlias

None

nemo_microservices.types.model_search.Peft: typing_extensions.TypeAlias

None

nemo_microservices.types.model_search.Prompt: typing_extensions.TypeAlias

None

nemo_microservices.types.model_search.Spec: typing_extensions.TypeAlias

None