nemo_rl.data.energon.multimodal.cookers.generic#

Module Contents#

Functions#

_decode_json_payload

_get_crude_payload

_infer_modality

_get_media_value

cook_conversation

Convert one crude JSON-plus-media sample into canonical SFT data.

Data#

API#

nemo_rl.data.energon.multimodal.cookers.generic._MEDIA_TYPES#

‘frozenset(…)’

nemo_rl.data.energon.multimodal.cookers.generic._decode_json_payload(value: Any) dict[str, Any]#
nemo_rl.data.energon.multimodal.cookers.generic._get_crude_payload(
sample: megatron.energon.CrudeSample,
) dict[str, Any]#
nemo_rl.data.energon.multimodal.cookers.generic._infer_modality(member: str) str#
nemo_rl.data.energon.multimodal.cookers.generic._get_media_value(
sample: megatron.energon.CrudeSample,
entry: dict[str, Any],
) Any#
nemo_rl.data.energon.multimodal.cookers.generic.cook_conversation(
sample: megatron.energon.CrudeSample,
) nemo_rl.data.energon.multimodal.types.CanonicalSFTSample#

Convert one crude JSON-plus-media sample into canonical SFT data.

nemo_rl.data.energon.multimodal.cookers.generic.__all__#

[‘cook_conversation’]