
    oi	                         S SK r S SKJrJr  S SKJrJr  S SKJr  S SK	J
r
   " S S\\ R                  5      r\ " S S	\5      5       rg)
    N)	dataclassfield)OptionalUnion)PromptTuningConfig)PeftTypec                   (    \ rS rSrSrSrSrSrSrSr	g)	MultitaskPromptTuningInit   TEXTRANDOMAVERAGE_SOURCE_TASKSEXACT_SOURCE_TASKONLY_SOURCE_SHARED N)
__name__
__module____qualname____firstlineno__r   r   r   r   r   __static_attributes__r       d/home/james-whalen/.local/lib/python3.13/site-packages/peft/tuners/multitask_prompt_tuning/config.pyr
   r
      s    DF1+-r   r
   c                      ^  \ rS rSr% \" \R                  SS0S9r\\\	4   \
S'   \" SSS0S9r\\	   \
S'   \" S	SS
0S9r\\   \
S'   \" SSS0S9r\\   \
S'   \" SSS0S9r\\   \
S'   U 4S jrSrU =r$ )MultitaskPromptTuningConfig$   helpzHow to initialize the prompt tuning parameters. Can be one of TEXT, RANDOM, AVERAGE_SOURCE_TASKS, EXACT_SOURCE_TASK, ONLY_SOURCE_SHARED.)defaultmetadataprompt_tuning_initNz|The path of source state dict. This is required when training the downstream target prompt from the pretrained source prompt"prompt_tuning_init_state_dict_pathr   z!source task id for initializationprompt_tuning_init_task   ranks	num_ranksznumber of tasks	num_tasksc                 L   > [         TU ]  5         [        R                  U l        g )N)super__post_init__r   MULTITASK_PROMPT_TUNING	peft_type)self	__class__s    r   r(   )MultitaskPromptTuningConfig.__post_init__<   s    !99r   )r*   )r   r   r   r   r   r
   r   r   r   str__annotations__r    r   r!   intr$   r%   r(   r   __classcell__)r,   s   @r   r   r   $   s    @E)009
A7<=  9>/
9&  .31PsGt-uXc]u$Q&'9JKIx}K$Q&BS9TUIx}U: :r   r   )enumdataclassesr   r   typingr   r   peft.tuners.prompt_tuningr   
peft.utilsr   r.   Enumr
   r   r   r   r   <module>r8      sD     ( " 8 
.TYY 
. :"4 : :r   