nemo_microservices.types.shared_params.parameter_efficient_finetuning_data#

Module Contents#

Classes#

ParameterEfficientFinetuningData

API#

class nemo_microservices.types.shared_params.parameter_efficient_finetuning_data.ParameterEfficientFinetuningData#

Bases: typing_extensions.TypedDict

finetuning_type: typing_extensions.Required[nemo_microservices.types.shared.finetuning_type.FinetuningType]#

None

The type of finetuning.

lora: nemo_microservices.types.shared_params.lora_finetuning_data.LoraFinetuningData#

None

Data about a LoRA fine-tuned model.

p_tuning: nemo_microservices.types.shared_params.p_tuning_finetuning_data.PTuningFinetuningData#

None

Data about a p-tuned model.