o Ni@slddlZddlmZmZddlmZmZddlmZddl m Z Gddde ej Z eGdd d eZdS) N) dataclassfield)OptionalUnion)PromptLearningConfig)PeftTypec@seZdZdZdZdZdS)PromptTuningInitTEXT SAMPLE_VOCABRANDOMN)__name__ __module__ __qualname__r r r rrT/home/ubuntu/.local/lib/python3.10/site-packages/peft/tuners/prompt_tuning/config.pyrsrcseZdZUdZeejddidZeee fe d<edddidZ e e e d<eddd idZ e e e d <eddd idZe ee d <fd dZZS)PromptTuningConfiga This is the configuration class to store the configuration of a [`PromptEmbedding`]. Args: prompt_tuning_init (Union[[`PromptTuningInit`], `str`]): The initialization of the prompt embedding. `TEXT` will initialize with your text. `SAMPLE_VOCAB` will initialize with randomly sampled tokens from the model's vocabulary. `RANDOM` will initialize with randomly sampled continuous, soft tokens (warning: sampled soft tokens may fall outside of embedding manifold) prompt_tuning_init_text (`str`, *optional*): The text to initialize the prompt embedding. Only used if `prompt_tuning_init` is `TEXT`. tokenizer_name_or_path (`str`, *optional*): The name or path of the tokenizer. Only used if `prompt_tuning_init` is `TEXT`. tokenizer_kwargs (`dict`, *optional*): The keyword arguments to pass to `AutoTokenizer.from_pretrained`. Only used if `prompt_tuning_init` is `TEXT`. helpz.How to initialize the prompt tuning parameters)defaultmetadataprompt_tuning_initNz[The text to use for prompt tuning initialization. Only used if prompt_tuning_init is `TEXT`prompt_tuning_init_textz`The tokenizer to use for prompt tuning initialization. Only used if prompt_tuning_init is `TEXT`tokenizer_name_or_pathzkThe keyword arguments to pass to `AutoTokenizer.from_pretrained`. Only used if prompt_tuning_init is `TEXT`tokenizer_kwargscsttj|_|jtjkr |js t dtjj d|jd|jtjkr9|j dur9t dtjj d|j d|j rL|jtjkrNt dtjj ddSdS)NzWhen prompt_tuning_init='z#', tokenizer_name_or_path can't be .z$', prompt_tuning_init_text can't be z;tokenizer_kwargs only valid when using prompt_tuning_init='z'.) super __post_init__r PROMPT_TUNING peft_typerrr r ValueErrorvaluerr)self __class__rrrKs&   z PromptTuningConfig.__post_init__)r r r__doc__rrr rrstr__annotations__rrrrdictr __classcell__rrr!rrs.  r)enum dataclassesrrtypingrr peft.configr peft.utilsrr$Enumrrrrrrs