nemo_microservices.types.shared_params.p_tuning_finetuning_data

Module Contents

Classes

PTuningFinetuningData

API

class nemo_microservices.types.shared_params.p_tuning_finetuning_data.PTuningFinetuningData

Bases: typing_extensions.TypedDict

token_embeddings: typing_extensions.Required[Iterable[Iterable[float]]]

None

Learned continuous prompt embeddings that help optimize the model for specific tasks.