bi^ddlZddlmZmZddlmZddlmZddlm Z Gdde ejZ eGdd eZ y) N) dataclassfield)Union)PromptLearningConfig)PeftTypeceZdZdZdZy)#PromptEncoderReparameterizationTypeMLPLSTMN)__name__ __module__ __qualname__r r V/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/peft/tuners/p_tuning/config.pyr r s C Drr ceZdZUdZeej ddiZee efe d<edddiZ e e d<ed dd iZ e e d <ed dd iZee d<fdZxZS)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_dropoutcLt|tj|_y)N)super __post_init__rP_TUNING peft_type)self __class__s rrz!PromptEncoderConfig.__post_init__:s !**r)r r r__doc__rr r rrstr__annotations__rintrrfloatr __classcell__)r"s@rrrs X]377DEX#U30S+S%T %AB $FG#=>OU ++rr)enum dataclassesrrtypingr peft.configr peft.utilsrr$Enumr rrrrr/sD (,#tyy  +.+ +r