nemo_automodel.shared.utils | NVIDIA NeMo AutoModel

Module Contents

Name	Description
`dtype_from_str`	Translate a str val of a dtype into the corresponding torch.dtype

nemo_automodel.shared.utils.dtype_from_str(
    val,
    default = torch.bfloat16
)

Translate a str val of a dtype into the corresponding torch.dtype Args: val (str): the dotted path of the dtype (e.g., “torch.bfloat16”).

Returns:

torch.dtype: the actual dtype (e.g., torch.bfloat16)