Zj' ZSrSSKJr SSKJr SSKJr /SQr\"SS\55rS/r g) z Processor class for EVOLLA. ) BatchFeature)ProcessorMixin)auto_docstring)aa_seqfoldseekmsac ^\rSrSrSU4SjjrSSjrSS\4Sjjr\SS\ \ \ -S-S\ \ \ \ \ -S-S \S-S\S-4S jj5r S r S r S rSrSrU=r$)EvollaProcessorNc >Uc [S5eUc [S5e[TU] X5 SURlX0lX@lg)aQ protein_tokenizer (`EsmTokenizer`): An instance of [`EsmTokenizer`]. The protein tokenizer is a required input. protein_max_length (`int`, *optional*, defaults to 1024): The maximum length of the sequence to be generated. text_max_length (`int`, *optional*, defaults to 512): The maximum length of the text to be generated. Nz+You need to specify an `protein_tokenizer`.z"You need to specify a `tokenizer`.z<|reserved_special_token_0|>) ValueErrorsuper__init__ tokenizer pad_tokenprotein_max_lengthtext_max_length)selfprotein_tokenizerrrrkwargs __class__s }/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/evolla/processing_evolla.pyrEvollaProcessor.__init__sO  $JK K  AB B *6#A "4.c P/nUHnURS5nURS5nSR[XV5VVs/sH&upxUR5UR 5-PM( snn5n UR U 5 M UR USSUSS9n U $s snnf)NrrptT)return_tensors truncation max_lengthpadding)getjoinzipupperlowerappendr) rproteinsr sa_sequencesproteinrrsf sa_sequence sa_tokenss rprocess_proteins EvollaProcessor.process_proteins2s G[[*F{{:.H''SEZ"[EZTQ1779qwwy#8EZ"[\K    ,  ** $K]gk+  #\s-B" rc /nUH/nURRUSSS9nURU5 M1 URUSSSSUS9nU$)NFT)tokenizeadd_generation_promptrlongest)add_special_tokensrr!rr )rapply_chat_templater')rtextsrpromptsmessagesprompt prompt_inputss r process_textEvollaProcessor.process_text?sr H^^77&*8F NN6 "  $& ' rr( messages_listrc JUbUc [S5eUbUO URnUbUO URn[U[5(aU/n[U[ [ 45(a![US[ [ 45(dU/n[U[ [ 45(a"[SU55(d [S5e[U[ [ 45(a;[SU55(d$[SSR[5SU35e[U[ [ 45(aUHn[U[ [ 45(d[S [U5S 35e[S U55(d [S 5e[S U55(d[SU55(dM[SU35e O[S[U5S 35eURX5nURX$5n[USUSUSUSS.S9$)a proteins (`Union[List[dict], dict]`): A list of dictionaries or a single dictionary containing the following keys: - `"aa_seq"` (`str`) -- The amino acid sequence of the protein. - `"foldseek"` (`str`) -- The foldseek string of the protein. messages_list (`Union[List[List[dict]], List[dict]]`): A list of lists of dictionaries or a list of dictionaries containing the following keys: - `"role"` (`str`) -- The role of the message. - `"content"` (`str`) -- The content of the message. protein_max_length (`int`, *optional*, defaults to 1024): The maximum length of the sequence to be generated. text_max_length (`int`, *optional*, defaults to 512): The maximum length of the text. Return: a dict with following keys: - `protein_input_ids` (`torch.Tensor` of shape `(batch_size, sequence_length)`) -- The input IDs for the protein sequence. - `protein_attention_mask` (`torch.Tensor` of shape `(batch_size, sequence_length)`) -- The attention mask for the protein sequence. - `text_input_ids` (`torch.Tensor` of shape `(batch_size, sequence_length)`) -- The input IDs for the text sequence. - `text_attention_mask` (`torch.Tensor` of shape `(batch_size, sequence_length)`) -- The attention mask for the text sequence. z3You need to specify `messages_list` and `proteins`.c3B# UHn[U[5v M g7fN isinstancedict.0ps r +EvollaProcessor.__call__..s:aX`ST:a;N;NX`zUThe proteins should be a list of dictionaries, but not all elements are dictionaries.c3b# UH%n[SUR555v M' g7f)c32# UH o[;v M g7frB)PROTEIN_VALID_KEYS)rGks rrI5EvollaProcessor.__call__...s:A''sN)allkeysrFs rrIrJs'; DLqC:: : :Hs-/z2There should be a list of dictionaries with keys: z, z for each protein.But got: z;Each messages in messages_list should be a list instead of .c3B# UHn[U[5v M g7frBrCrGms rrIrJsA1:a..rKzfEach message in messages_list should be a list of dictionaries, but not all elements are dictionaries.c3Z# UH!n[UR55S:gv M# g7f)N)lenrRrUs rrIrJs<8as1668})8s)+c3^# UH#n[UR55SS1:gv M% g7f)rolecontentN)setrRrUs rrIrJs(DBJQCMfi%88(s+-zlEach message in messages_list should be a list of dictionaries with two keys: 'role' and 'content'.But got: zFThe messages_list should be a list of lists of dictionaries, but it's input_idsattention_mask)protein_input_idsprotein_attention_maskr^r_)data)r rrrDrElisttuplerQr#rN TypeErrortypeanyr/r<r) rr(r>rrrr9r. text_tokenss r__call__EvollaProcessor.__call__WsJ>  }4RS S3E3Q/W[WnWn-<-H/dNbNb h % % zH mdE] 3 3J}UVGWZ^`eYf"?   rc:URR"U0UD6$rB)r batch_decoderargsrs rrlEvollaProcessor.batch_decodes~~**D;F;;rc:URR"U0UD6$rB)rdecoderms rrqEvollaProcessor.decodes~~$$d5f55rc:URR"U0UD6$rB)rrlrms rprotein_batch_decode$EvollaProcessor.protein_batch_decodes%%22DCFCCrc:URR"U0UD6$rB)rrqrms rprotein_decodeEvollaProcessor.protein_decodes%%,,d=f==r)rr)N)ry)rz)NNNN)__name__ __module__ __qualname____firstlineno__rr/intr<rrcrErirlrqrtrw__static_attributes__ __classcell__)rs@rr r s/(  #0.2>B)-&* T t*t#d*T DJ'$t*4t;T  $J T t T T l<6D>>rr N) __doc__feature_extraction_utilsrprocessing_utilsrutilsrrNr __all__rrrsJ5$3[>n[>[>|  r