Skip to navigation
ReferenceFull Library ReferenceNemo AutomodelNemo AutomodelComponentsModelsD1nemo_automodel.components.models.d1.releases

nemo_automodel.components.models.d1.releases

View as Markdown

Pinned publisher sources and execution precision for supported native D1 releases.

Module Contents

Classes

NameDescription
D1ReleaseIdentify one immutable publisher implementation and checkpoint.

Data

RELEASES

RELEASE_DTYPES

API

class nemo_automodel.components.models.d1.releases.D1Release()

Bases: NamedTuple

Identify one immutable publisher implementation and checkpoint.

reference_class
str
repository
str
revision
str
nemo_automodel.components.models.d1.releases.RELEASES: dict[str, D1Release] = {'d1-3b': D1Release('LiquidAI/d1-3B', '051bcc464b01b9f92942b364d9586b0ef5912432'...
 
nemo_automodel.components.models.d1.releases.RELEASE_DTYPES: dict[str, dtype] = {'d1-3b': torch.bfloat16, 'd1-omni': torch.float32}