o Ni^@shddlZddlmZmZddlmZddlmZddlm Z Gddde ej Z eGdd d eZ dS) N) dataclassfield)Union)PromptLearningConfig)PeftTypec@seZdZdZdZdS)#PromptEncoderReparameterizationTypeMLPLSTMN)__name__ __module__ __qualname__rr r r O/home/ubuntu/.local/lib/python3.10/site-packages/peft/tuners/p_tuning/config.pyrsrcseZdZUdZeejddidZee efe d<edddidZ e e d<ed dd idZ e e d <ed dd idZee d<fddZZS)PromptEncoderConfiga This is the configuration class to store the configuration of a [`PromptEncoder`]. Args: encoder_reparameterization_type (Union[[`PromptEncoderReparameterizationType`], `str`]): The type of reparameterization to use. encoder_hidden_size (`int`): The hidden size of the prompt encoder. encoder_num_layers (`int`): The number of layers of the prompt encoder. encoder_dropout (`float`): The dropout probability of the prompt encoder. helpz(How to reparameterize the prompt encoder)defaultmetadataencoder_reparameterization_typeNz%The hidden size of the prompt encoderencoder_hidden_sizez*The number of layers of the prompt encoderencoder_num_layersgz!The dropout of the prompt encoderencoder_dropoutcsttj|_dS)N)super __post_init__rP_TUNING peft_type)self __class__r rr:s  z!PromptEncoderConfig.__post_init__)r r r __doc__rrrrrstr__annotations__rintrrfloatr __classcell__r r rrrs&  r)enum dataclassesrrtypingr peft.configr peft.utilsrr Enumrrr r r rs