nemo_automodel.shared.utils

Module Contents

Functions

dtype_from_str

Translate a str val of a dtype into the corresponding torch.dtype

API

nemo_automodel.shared.utils.dtype_from_str(val)

Translate a str val of a dtype into the corresponding torch.dtype

Parameters:

val (str) – the dotted path of the dtype (e.g., “torch.bfloat16”).

Returns:

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

Return type:

torch.dtype