§
     ’ÞjPF  ã                   óÌ   — d dl Z d dlZd dlZd dlZddlmZ dZdddddd	d
œZd!d„Zd„ Z	d„ Z
d„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zddœd„Zd„ Zd„ Zdddddddœd „ZdS )"é    Né   )Únormalize_specialitiesÚmetadatazText/new generationzStart imagez	End imagezContinue from source videoz"Continue from last generated video)Ú ÚTÚSÚEÚVÚLFc                 óž   — |r|                      dd ¦  «        }|�||v r|S |                      | d ¦  «        }|€dS |                     ¦   «         S )NÚgroupÚunknown)ÚgetÚquery_model_family)Úbase_model_typeÚ	model_defÚmodel_types_handlersÚfamilies_infosÚfor_uiÚmodel_familyÚhandlers          ú=C:\Users\johnn\OneDrive\Documents\AI\models\model_metadata.pyÚget_model_familyr      sd   € Øð  Ø —}’} W¨dÑ3Ô3ˆØÐ#¨¸Ð(FÐ(FØÐØ"×&Ò& ¸Ñ=Ô=€GØ€ØˆyØ×%Ò%Ñ'Ô'Ð'ó    c           
      óì   — t           j                             t           j                             t	          |                      dd¦  «        pd¦  «        ¦  «        ¦  «                             ¦   «         dk    S )NÚpathr   Ú	finetunes)Úosr   ÚbasenameÚdirnameÚstrr   Úcasefold©r   s    r   Ú_model_def_finetuner$   !   sS   € ÝŒ7×Ò�BœGŸOšO­C°	·²¸fÀbÑ0IÔ0IÐ0OÈRÑ,PÔ,PÑQÔQÑRÔR×[Ò[Ñ]Ô]ÐalÒlÐlr   c                 ó†  — t          | t          ¦  «        sg S g }|                      dg ¦  «        pg D ]�}t          |t          t          f¦  «        r[|                     t          t          |¦  «        dk    r|d         nt          |¦  «        dk    r|d         nd¦  «        pd¦  «         Œy|                     t          |pd¦  «        ¦  «         Œž|                     d„ |                      dg ¦  «        pg D ¦   «         ¦  «         |                      dd ¦  «        }|�$|                     t          |pd¦  «        ¦  «         |S )NÚchoicesr   r   r   c              3   ó8   K  — | ]}t          |pd ¦  «        V — ŒdS )r   N)r!   ©Ú.0Úvalues     r   ú	<genexpr>z!_choice_values.<locals>.<genexpr>.   s.   è è € ÐVÐV u•#�e�k˜rÑ"Ô"ÐVÐVÐVÐVÐVÐVr   Ú	selectionÚdefault)	Ú
isinstanceÚdictr   ÚlistÚtupleÚappendr!   ÚlenÚextend)Ú
choice_defÚvaluesÚchoicer-   s       r   Ú_choice_valuesr8   %   s:  € Ý�j¥$Ñ'Ô'ð Øˆ	Ø€FØ—.’. ¨BÑ/Ô/Ð5°2ð -ð -ˆÝ�f�t¥U˜mÑ,Ô,ð 	-Ø�MŠM�#­3¨v©;¬;¸ª?¨?˜f Qœi˜iÍSÐQWÉ[Ì[Ð[\Ê_È_ÀÀqÄ	À	ÐbdÑeÔeÐkÐikÑlÔlÐlÐlà�MŠM�#˜f˜l¨Ñ+Ô+Ñ,Ô,Ð,Ð,Ø
‡M‚MÐVÐV°
·²¸{ÈBÑ0OÔ0OÐ0UÐSUÐVÑVÔVÑVÔVÐVØ�nŠn˜Y¨Ñ-Ô-€GØÐØ�Š•c˜'˜- RÑ(Ô(Ñ)Ô)Ð)Ø€Mr   c                 óT   ‡— t          ˆfd„t          | ¦  «        D ¦   «         ¦  «        S )Nc              3   óP   •‡K  — | ]Št          ˆfd „‰D ¦   «         ¦  «        V — Œ dS )c              3   ó    •K  — | ]}|‰v V — Œ	d S ©N© )r)   Úletterr*   s     €r   r+   z3_choice_values_contain.<locals>.<genexpr>.<genexpr>6   s'   øè è € Ð9Ð9 v�6˜U�?Ð9Ð9Ð9Ð9Ð9Ð9r   N)Úany)r)   r*   Úletterss    @€r   r+   z)_choice_values_contain.<locals>.<genexpr>6   sA   øøè è € ÐbÐb¸e�sÐ9Ð9Ð9Ð9°Ð9Ñ9Ô9Ñ9Ô9ÐbÐbÐbÐbÐbÐbr   )r?   r8   )r5   r@   s    `r   Ú_choice_values_containrA   5   s/   ø€ ÝÐbÐbÐbÐbÅ~ÐV`ÑGaÔGaÐbÑbÔbÑbÔbÐbr   c                 óâ  ‡‡— t          | t          ¦  «        sg S |                      di ¦  «        pi }g Št          ¦   «         Šˆˆfd„}|                      dg ¦  «        pg D ]…}t          |t          t
          f¦  «        rGt          |¦  «        dk    r|d         nd}t          |¦  «        dk    r|d         n|} |||¦  «         Œe ||                     ||¦  «        |¦  «         Œ†|                     ¦   «         D ]9\  }} |||                     dd¦  «                             dd¦  «        ¦  «         Œ:|                      d	g ¦  «        pg D ]"} ||                     ||¦  «        |¦  «         Œ#‰S )
NÚlabelsc                 ó¶   •— |€dnt          |¦  «        }|‰v rd S ‰                     |¦  «         ‰                     t          | p|pd¦  «        |dœ¦  «         d S )Nr   ÚNone©Úlabelr*   )r!   Úaddr2   )rG   r*   r&   Úseens     €€r   Ú
add_choicez'_normalized_choices.<locals>.add_choice@   se   ø€ Ø�m��­¨U©¬ˆØ�Dˆ=ˆ=ØˆFØ�Š�‰ŒˆØ�Š¥ UÐ%=¨eÐ%=°vÑ!>Ô!>ÈÐOÐOÑPÔPÐPÐPÐPr   r&   r   r   r   r
   ÚPr,   )	r.   r/   r   Úsetr0   r1   r3   ÚitemsÚreplace)	r5   rC   rJ   r7   rG   r*   Úkeyr&   rI   s	          @@r   Ú_normalized_choicesrP   9   s§  øø€ Ý�j¥$Ñ'Ô'ð Øˆ	Ø�^Š^˜H bÑ)Ô)Ð/¨R€FØ€GÝ‰5Œ5€DðQð Qð Qð Qð Qð Qð —.’. ¨BÑ/Ô/Ð5°2ð ;ð ;ˆÝ�f�t¥U˜mÑ,Ô,ð 	;Ý!$ V¡¤¨q¢ �F˜1”I�I°bˆEÝ!$ V¡¤¨q¢ �F˜1”I�I°eˆEØˆJ�u˜eÑ$Ô$Ð$Ð$àˆJ�v—z’z &¨&Ñ1Ô1°6Ñ:Ô:Ð:Ð:Ø—l’l‘n”nð Að A‰
ˆˆUØˆ
�5˜#Ÿ+š+ c¨2Ñ.Ô.×6Ò6°s¸BÑ?Ô?Ñ@Ô@Ð@Ð@Ø—’ ¨RÑ0Ô0Ð6°Bð 4ð 4ˆØˆ
�6—:’:˜e UÑ+Ô+¨UÑ3Ô3Ð3Ð3Ø€Nr   c                 óz   ‡ — t          ‰ t          ¦  «        sd S d}ˆ fd„|D ¦   «         }t          ‰ ¦  «        |d<   |S )N)r-   rG   Úletters_filterÚvisibleÚ
show_labelÚscaleÚtriggerÚtypec                 ó*   •— i | ]}|‰v ¯|‰|         “ŒS r=   r=   )r)   rO   r5   s     €r   ú
<dictcomp>z(normalize_choice_def.<locals>.<dictcomp>Y   s)   ø€ ÐLÐLÐL¨3¸#ÀÐ:KÐ:K�#�z #”Ð:KÐ:KÐ:Kr   r&   )r.   r/   rP   )r5   ÚkeysÚ
normalizeds   `  r   Únormalize_choice_defr\   U   sP   ø€ Ý�j¥$Ñ'Ô'ð ØˆtØf€DØLÐLÐLÐL°$ÐLÑLÔL€JÝ/°
Ñ;Ô;€JˆyÑØÐr   c                 ó2  — |                       dd¦  «        rddgS |                       dd¦  «        rdgS |                       dd¦  «        rdgS |                       d	d¦  «        rd
gS |                       dd¦  «        s|                       dd¦  «        rd
dgS dgS )NÚworld_outputsFÚworldÚvideoÚ
audio_onlyÚaudioÚmesh_outputsÚ3dÚimage_outputsÚimageÚv2i_switch_supportedÚinpaint_support)r   r#   s    r   Úinfer_main_outputsri   ^   s½   € Ø‡}‚}�_ eÑ,Ô,ð "Ø˜Ð!Ð!Ø‡}‚}�\ 5Ñ)Ô)ð ØˆyÐØ‡}‚}�^ UÑ+Ô+ð ØˆvˆØ‡}‚}�_ eÑ,Ô,ð ØˆyÐØ‡}‚}Ð+¨UÑ3Ô3ð "°y·}²}ÐEVÐX]Ñ7^Ô7^ð "Ø˜Ð!Ð!Øˆ9Ðr   c                 ó‚   — t          | ¦  «        }|                      dd¦  «        rd|vr|                     d¦  «         |S )NÚreturns_audioFrb   )ri   r   r2   )r   Úoutputss     r   Úinfer_outputsrm   l   sF   € Ý  Ñ+Ô+€GØ‡}‚}�_ eÑ,Ô,ð  °ÀÐ1GÐ1GØ�Š�wÑÔÐØ€Nr   c                 óŒ  — dg}t          |                      dd¦  «        pd¦  «        }|                      dd ¦  «        }|                      dd ¦  «        }|                      dd ¦  «        }|                      dd ¦  «        }t          |                      dd	¦  «        ¦  «        }|                      d
d	¦  «        r|                     d¦  «         d|v sPd|v sL|                      dd	¦  «        s6|                      dd	¦  «        s t	          |d¦  «        st	          |d¦  «        r|                     d¦  «         |rO|                      dd	¦  «        s9t	          |d¦  «        st	          |d¦  «        rd|vr|                     d¦  «         |                      dd	¦  «        s:d|v s6d|v s2|sEt	          |d¦  «        s t	          |d¦  «        st	          |d¦  «        r|                     d¦  «         t          | ¦  «        r"|                     t          | ¦  «        ¦  «         t          t           	                    |¦  «        ¦  «        S )NÚtextÚimage_prompt_types_allowedr   Úimage_ref_choicesÚguide_custom_choicesÚguide_preprocessingÚcustom_video_selectionre   FÚany_audio_promptrb   r   r	   Úend_frames_always_enabledrh   ÚIKFrf   Úreference_video_enabledr
   r   r`   )
r!   r   Úboolr2   rA   Úcustom_guide_kindsr4   r0   r/   Úfromkeys)r   Úinputsrp   Ú
image_refsÚalt_guide_refsrs   rt   re   s           r   Úinfer_inputsr   s   sA  € ØˆX€FÝ!$ Y§]¢]Ð3OÐQSÑ%TÔ%TÐ%ZÐXZÑ![Ô![ÐØ—’Ð2°DÑ9Ô9€JØ—]’]Ð#9¸4Ñ@Ô@€NØ#Ÿ-š-Ð(=¸tÑDÔDÐØ&Ÿ]š]Ð+CÀTÑJÔJÐÝ˜Ÿš ¸Ñ>Ô>Ñ?Ô?€Mà‡}‚}Ð'¨Ñ/Ô/ð Ø�Š�gÑÔÐØ
Ð(Ð(Ð(¨CÐ3MÐ,MÐ,MÐQZ×Q^ÒQ^Ð_zð  }Bñ  RCô  RCÐ,Mð  GP÷  GTò  GTð  Ufð  hmñ  Gnô  GnÐ,Mõ  rHð  ISð  UZñ  r[ô  r[Ð,Mõ  _uð  vDð  FKñ  _Lô  _LÐ,MØ�Š�gÑÔÐàð ˜YŸ]š]Ð+DÀeÑLÔLð ÕRhÐi|ð  Bñ  SCô  SCð õ  G]ð  ^lð  nqñ  Grô  Grð ð  xð  GMð  xMð  xMØ�Š�gÑÔÐØ‡}‚}Ð.°Ñ6Ô6ð ¸#ÐA[Ð:[Ð:[Ð_bð  gAð  `Að  `Að  JWð  `Aõ  ]sð  tGð  ILñ  ]Mô  ]Mð  `Aõ  Qgð  hvð  x{ñ  Q|ô  Q|ð  `Aõ  @Vð  Wmð  orñ  @sô  @sð  `AØ�Š�gÑÔÐÝ˜)Ñ$Ô$ð 5Ø�ŠÕ(¨Ñ3Ô3Ñ4Ô4Ð4Ý•—’˜fÑ%Ô%Ñ&Ô&Ð&r   c                 óâ   ‡‡— ddl mŠmŠ |                      d¦  «        }|€g S ˆˆfd„|                     dg ¦  «        D ¦   «         }t	          t
                               |pdg¦  «        ¦  «        S )zqMedia kinds of the custom_guide file input ("3d", "world", ...; "file" for other formats such as MIDI or scores).r   )ÚMESH_IMPORT_EXTENSIONSÚWORLD_EXTENSIONSÚcustom_guideNc                 ót   •— g | ]4}|                      ¦   «         ‰v rd n|                      ¦   «         ‰v rdnd‘Œ5S )rd   r_   Úfile)Úlower)r)   Ú	file_typer�   r‚   s     €€r   ú
<listcomp>z&custom_guide_kinds.<locals>.<listcomp>�   sj   ø€ ð  xð  xð  xð  HQ�Y—_’_Ñ&Ô&Ð*@Ð@Ð@ˆTˆTÐQZ×Q`ÒQ`ÑQbÔQbÐfvÐQvÐQvÀgÀgð  }Cð  xð  xð  xr   Ú
file_typesr…   )Úshared.utils.gallery_sourcesr�   r‚   r   r0   r/   r{   )r   rƒ   Úkindsr�   r‚   s      @@r   rz   rz   Š   s¼   øø€ àUÐUÐUÐUÐUÐUÐUÐUØ—=’= Ñ0Ô0€LØÐØˆ	ð xð  xð  xð  xð  xð  Ua÷  Ueò  Ueð  frð  tvñ  Uwô  Uwð  xñ  xô  x€EÝ•—’˜eÐ/¨ xÑ0Ô0Ñ1Ô1Ð1r   c                 óx  — t          |                      dd¦  «        pd¦  «        }|                      dd ¦  «        }|                      dd ¦  «        }|                      dd ¦  «        }|                      dd ¦  «        }d„ t          |¦  «        t          |¦  «        z   D ¦   «         }t          d„ |D ¦   «         ¦  «        }t	          |                      d	d
¦  «        p|                      dd
¦  «        ¦  «        }t          |d¦  «        pt          |d¦  «        pt          |d¦  «        }	t	          |                      dd
¦  «        ¦  «        }
|
o|                      dd
¦  «         }|                      d¦  «        }i |�ld|                     dd¦  «        t          | ¦  «        t          |                     dg ¦  «        ¦  «        t	          |                     dd
¦  «        ¦  «        dœini ¥d|v d|v p"t	          |                      dd
¦  «        ¦  «        ||o||o| t          d„ |D ¦   «         ¦  «        t          |d¦  «        pt          |d¦  «        |o|	|
oFt          |                      dd ¦  «        d¦  «        p"t	          |                      dd
¦  «        ¦  «        dœ	t	          |                      dd
¦  «        ¦  «        d|v d|v |
 o|	|
 o#t          |                      dd ¦  «        d¦  «        d œt	          |                      d!d
¦  «        ¦  «        t	          |                      d"d
¦  «        ¦  «        t	          |                      d#d
¦  «        ¦  «        p#t          |                      d$d ¦  «        d¦  «        t	          |                      d%d
¦  «        p|                      d&d
¦  «        ¦  «        d'œd(œ¥S ))Nrp   r   rq   rs   rr   rt   c                 ó   — g | ]}d |v¯|‘Œ	S )ÚFr=   r(   s     r   rˆ   z&infer_media_inputs.<locals>.<listcomp>š   sE   € ð  Cð  Cð  C 5Ðqtð  }Bð  rBð  rB˜ð  rBð  rBð  rBr   c              3   ó   K  — | ]}d |v V — Œ	dS )ÚINr=   r(   s     r   r+   z%infer_media_inputs.<locals>.<genexpr>›   s&   è è € ÐDÐD¨˜˜u˜ÐDÐDÐDÐDÐDÐDr   Úone_image_ref_neededFÚone_image_ref_onlyr
   re   rx   rƒ   Úcustom_fileÚnamezCustom Guider‰   Úrequired)r”   r‹   r‰   r•   r   r	   rv   c              3   ó   K  — | ]}d |v V — Œ	dS )ÚKNr=   r(   s     r   r+   z%infer_media_inputs.<locals>.<genexpr>©   s&   è è € ÐJÐJ¨u˜c U˜lÐJÐJÐJÐJÐJÐJr   rŽ   Úmask_preprocessingÚArh   )	ÚstartÚendÚ	referenceÚsingle_referenceÚmultiple_referencesÚ
backgroundÚinjected_framesÚcontrolÚmaskr   )rœ   ÚcontinueÚlastr¡   r¢   ru   Úreference_audio_enabledÚoutput_audio_is_input_audioÚaudio_prompt_type_sourcesra   rk   )Úpromptrœ   r¡   Úoutput)rf   r`   rb   )r!   r   r8   r?   ry   rA   rz   r0   )r   rp   r}   rs   rr   rt   Úreference_choicesÚhas_referencer�   Úhas_controlre   Úimage_controlrƒ   s                r   Úinfer_media_inputsr®   ”   sÃ  € Ý!$ Y§]¢]Ð3OÐQSÑ%TÔ%TÐ%ZÐXZÑ![Ô![ÐØ—’Ð2°DÑ9Ô9€JØ#Ÿ-š-Ð(=¸tÑDÔDÐØ$Ÿ=š=Ð)?ÀÑFÔFÐØ&Ÿ]š]Ð+CÀTÑJÔJÐð Cð  C­N¸:Ñ,FÔ,FÍÐXlÑImÔImÑ,mð  Cñ  Cô  CÐÝÐDÐDÐ2CÐDÑDÔDÑDÔD€MÝ˜IŸMšMÐ*@À%ÑHÔHÐvÈIÏMÊMÐZnÐpuÑLvÔLvÑwÔwÐÝ(Ð)<¸cÑBÔBð  oÕF\Ð]qÐsvÑFwÔFwð  oõ  |Rð  Sið  knñ  |oô  |o€KÝ˜Ÿš ¸Ñ>Ô>Ñ?Ô?€MØ!ÐY¨)¯-ª-Ð8QÐSXÑ*YÔ*YÐ&Y€MØ—=’= Ñ0Ô0€Lðð jvð  jBˆM L×$4Ò$4°V¸^Ñ$LÔ$LÕWiÐjsÑWtÔWtõ  EIð  JV÷  JZò  JZð  [gð  ikñ  Jlô  Jlñ  Emô  Emõ  {ð  @L÷  @Pò  @Pð  Q[ð  ]bñ  @cô  @cñ  {dô  {dð  eð  eð  fð  fð  HJðð Ð6Ð6ØÐ4Ð4Ðo½¸Y¿]º]ÐKfÐhmÑ=nÔ=nÑ8oÔ8oØ&Ø -Ð BÐ2BØ#0Ð#IÐ9IÐ5IÝÐJÐJÐ8IÐJÑJÔJÑJÔJÝ5°jÀ#ÑFÔFÐ{ÕJ`ÐauÐwzÑJ{ÔJ{Ø$Ð4¨Ø!ð  `Õ'=¸i¿mºmÐL`ÐbfÑ>gÔ>gÐilÑ'mÔ'mð  (_ÕquÐv÷  wDò  wDð  EVð  X]ñ  w^ô  w^ñ  r_ô  r_ð

ð 

õ ˜iŸmšmÐ,EÀuÑMÔMÑNÔNØÐ9Ð9ØÐ5Ð5Ø)Ð)Ð:¨{Ø&Ð&ÐrÕ,BÀ9Ç=Â=ÐQeÐgkÑClÔClÐnqÑ,rÔ,rð
ð 
õ ˜9Ÿ=š=Ð);¸UÑCÔCÑDÔDÝ˜iŸmšmÐ,EÀuÑMÔMÑNÔNå˜IŸMšMÐ*GÈÑOÔOÑPÔPð  bÕTjÐkt×kxÒkxð  zUð  W[ñ  l\ô  l\ð  ^añ  Ubô  UbÝ˜9Ÿ=š=¨°uÑ=Ô=ÐfÀÇÂÈÐ`eÑAfÔAfÑgÔgð
ð 
ð)ð ð ð r   c           	      óÚ  — |d         }|d         }|d         }i dd|v od|v “dd|v o|d         “dd|v o|d	         p|d
         p|d         “dd|v od|v o|                       dd¦  «         “dd|v o|d         p|d         p|d
         “dd|v od|v “dd|v o|d         “dd|v o|d         “dd|v “dt          |                       dd¦  «        p|d         p|d         ¦  «        “dt          |                       dd¦  «        ¦  «        “d|d         “d|d         “d|d         “d|d         “d|d
         “d |d
         “|d	         t          |                       d!d¦  «        ¦  «        t          |                       d"d¦  «        ¦  «         d#œ¥d$|v r.|d         d|v o|                       dd¦  «         |d         d$|v d%œni ¥d&|v rd|v |d         d'œni ¥|                       d(i ¦  «        ¥S ))Nrf   r`   rb   Útext_to_videoro   Úimage_to_videorš   Úvideo_to_videor£   r¡   rœ   Útext_to_imager‘   FÚimage_to_imageÚtext_to_audioÚaudio_to_audior¨   Úaudio_to_videoÚaudio_outputÚ
inpaintingrh   r¢   ÚoutpaintingÚvideo_guide_outpaintingÚreference_imagesÚreference_videosÚbackground_imagerŸ   r    Úcontrol_imageÚcontrol_videoÚsliding_windowÚno_lora)Úvideo_continuationrÁ   Úlorard   )Úimage_to_3dÚ
text_to_3dÚvideo_to_3dÚ3d_to_3dr_   )Útext_to_worldÚimage_to_worldÚcapability_overrides)r   ry   )r   Úmain_outputsrl   r|   Úmedia_inputsÚimage_inputsÚvideo_inputsÚaudio_inputss           r   Úinfer_capabilitiesrÑ   ¿   sy  € Ø Ô(€LØ Ô(€LØ Ô(€LðØ˜ LÐ0ÐE°V¸vÐ5Eðà˜' \Ð1ÐK°lÀ7Ô6Kðð 	˜' \Ð1ð  I°|ÀJÔ7Oð  8HÐS_Ð`iÔSjð  8HÐnzð  |Gô  oHðð 	˜ LÐ0Ðz°V¸vÐ5EÐzÈiÏmÊmÐ\rÐtyÑNzÔNzÐJzð	ð
 	˜' \Ð1ð  F°|ÀGÔ7Lð  8EÐP\Ð]hÔPið  8EÐmyð  {Dô  nEðð 	˜ LÐ0ÐE°V¸vÐ5Eðð 	˜' \Ð1ÐL°lÀ8Ô6Lðð 	˜' \Ð1ÐL°lÀ8Ô6Lðð 	˜ 7Ð*ðð 	•d˜9Ÿ=š=Ð):¸EÑBÔBÐrÀlÐSYÔFZÐrÐ^jÐkqÔ^rÑsÔsðð 	•t˜IŸMšMÐ*CÀUÑKÔKÑLÔLðð 	˜L¨Ô5ðð 	˜L¨Ô5ðð 	˜L¨Ô6ðð 	˜<Ð(9Ô:ðð  	˜ iÔ0ð!ð" 	˜ iÔ0ð#ð$ +¨:Ô6Ý˜yŸ}š}Ð-=¸uÑEÔEÑFÔFÝ˜Ÿš y°%Ñ8Ô8Ñ9Ô9Ð9ð)ð ð ð, UYð  ]ið  Uið  Ui˜<¨Ô4ÀFÈfÐDTð  EJÐ]f×]jÒ]jð  lBð  DIñ  ^Jô  ^Jð  ZJð  [gð  hsô  [tð  BFð  JPð  BPð  Qð  Qð  Qð  oqð-ð. \cÐfrÐ[rÐ[r˜V vÐ-ÀÈgÔAVÐWÐWÐWÐxzð/ð2 �-Š-Ð.°Ñ
3Ô
3ð3ð r   c                 ó  — ddl m} t          |                      dd¦  «        pd¦  «        }t                               dd¦  «        ddœg}|D ]:}|dk    rŒ	|                     t                               ||¦  «        |dœ¦  «         Œ;||dœt          |                      dd ¦  «        ¦  «        t          |                      d	d ¦  «        ¦  «        t          |                      d
d ¦  «        ¦  «        t          |                      dd ¦  «        ¦  «        t          |                      dd ¦  «        ¦  «        t          |                      dd¦  «        pd¦  «        dœt          |                      dd ¦  «        ¦  «        |                      dd ¦  «        dœt          |                      dd ¦  «        ¦  «        |                      dd ¦  «        �%t          d|                      dg ¦  «        i¦  «        nd t           || ¦  «        ¦  «        dœS )Nr   )Úresolve_definitionrp   r   rF   r   )Úallowedr&   rs   r˜   rr   rq   rt   Úset_video_prompt_type)rs   r˜   rr   rq   rt   Úforcedr§   Úaudio_prompt_type_custom_option)ÚsourcesÚcustom_optionÚmodel_modesÚsample_solversr&   )Úimage_prompt_typeÚvideo_prompt_typeÚaudio_prompt_typeÚ
model_modeÚsample_solverÚprompt_enhancer)Úshared.prompt_enhancer.labelsrÓ   r!   r   Ú_IMAGE_PROMPT_LABELSr2   r\   )r   rÓ   rp   Úimage_prompt_choicesr>   s        r   Úinfer_setting_valuesrå   à   s<  € Ø@Ð@Ð@Ð@Ð@Ð@å!$ Y§]¢]Ð3OÐQSÑ%TÔ%TÐ%ZÐXZÑ![Ô![ÐÝ&:×&>Ò&>¸rÀ2Ñ&FÔ&FÐQSÐTÐTÐUÐØ,ð jð jˆØ�SŠ=ˆ=ØØ×#Ò#Õ.B×.FÒ.FÀvÈvÑ.VÔ.VÐagÐ$hÐ$hÑiÔiÐiÐið 2Ø+ð
ð 
õ
 $8¸	¿ºÐF[Ð]aÑ8bÔ8bÑ#cÔ#cÝ"6°y·}²}ÐEYÐ[_Ñ7`Ô7`Ñ"aÔ"aÝ$8¸¿ºÐG]Ð_cÑ9dÔ9dÑ$eÔ$eÝ!5°i·m²mÐDWÐY]Ñ6^Ô6^Ñ!_Ô!_Ý&:¸9¿=º=ÐIaÐcgÑ;hÔ;hÑ&iÔ&iÝ˜)Ÿ-š-Ð(?ÀÑDÔDÐJÈÑKÔKð
ð 
õ ,¨I¯MªMÐ:UÐW[Ñ,\Ô,\Ñ]Ô]Ø&Ÿ]š]Ð+LÈdÑSÔSð
ð 
õ +¨9¯=ª=¸ÈÑ+MÔ+MÑNÔNØcl×cpÒcpð  rBð  DHñ  dIô  dIð  dUÕ-¨y¸)¿-º-ÐHXÐZ\Ñ:]Ô:]Ð.^Ñ_Ô_Ð_ð  [_Ý/Ð0BÐ0BÀ9Ñ0MÔ0MÑNÔNð'ð ð r   c                 óf  ‡— ddl m} d‰v rt          ‰d         ¦  «        ‰d<   ‰                     dd ¦  «        p| }t	          |‰||d¬¦  «        }|                     ||                     dd¦  «        ¦  «        d	         }t          ‰¦  «        }t          ‰¦  «        }	t          ‰¦  «        }
t          ‰¦  «        }| |||t          ‰¦  «        ||	|
|t          ‰||	|
|¦  «        t          ‰¦  «        d
œt          d„ |D ¦   «         ¦  «        r |‰¦  «        ni ¥ˆfd„dD ¦   «         ¥‰t          <   ‰S )Nr   )Úresolution_metadataÚspecialitiesÚarchitectureT)r   r   )éd   ÚUnknownr   )Ú
model_typeÚfamilyÚfamily_labelr   ÚfinetuneÚmain_outputrl   r|   rÍ   ÚcapabilitiesÚsetting_valuesc              3   ó   K  — | ]}|d v V — Œ	dS ))rf   r`   Nr=   )r)   r©   s     r   r+   z!store_metadata.<locals>.<genexpr>  s)   è è € Ð0iÐ0iÐRX°Ð;MÐ1MÐ0iÐ0iÐ0iÐ0iÐ0iÐ0ir   c                 óN   •— i | ]!}|‰v ¯|t          j        ‰|         ¦  «        “Œ"S r=   )ÚcopyÚdeepcopy)r)   rO   r   s     €r   rY   z"store_metadata.<locals>.<dictcomp>  s5   ø€ Ð
uÐ
uÐ
u°#ÐdgÐktÐdtÐdtˆ3•”˜i¨œnÑ-Ô-ÐdtÐdtÐdtr   )ÚacceleratedÚsizerè   )Úshared.resolutionsrç   r   r   r   ri   rm   r   r®   r$   rÑ   rå   r?   ÚMETADATA_KEY)rì   r   r   r   rç   r   rí   rî   rÌ   rl   r|   rÍ   s    `          r   Ústore_metadatarû      s~  ø€ Ø6Ð6Ð6Ð6Ð6Ð6Ø˜Ð"Ð"Ý$:¸9À^Ô;TÑ$UÔ$Uˆ	�.Ñ!Ø—m’m N°DÑ9Ô9ÐG¸Z€OÝ˜o¨yÐ:NÐP^ÐgkÐlÑlÔl€FØ!×%Ò% f¨n×.@Ò.@ÀÐL\Ñ.]Ô.]Ñ^Ô^Ð_`Ôa€LÝ% iÑ0Ô0€LÝ˜IÑ&Ô&€GÝ˜)Ñ$Ô$€FÝ% iÑ0Ô0€Là ØØ$Ø*Ý'¨	Ñ2Ô2Ø#ØØØ$Ý*¨9°lÀGÈVÐUaÑbÔbÝ.¨yÑ9Ô9ðð õ .1Ð0iÐ0iÐ\hÐ0iÑ0iÔ0iÑ-iÔ-iÐqÐÐ˜yÑ)Ô)Ð)Ðoqðð vÐ
uÐ
uÐ
uÐ9`Ð
uÑ
uÔ
uð€I�lÑð Ðr   T©Úsplit_stringc                óh  — | €d S t          | t          t          t          f¦  «        rd„ | D ¦   «         }n||r5t          | t          ¦  «        r d„ t          j        d| ¦  «        D ¦   «         }nEt	          | ¦  «                             ¦   «         r"t	          | ¦  «                             ¦   «         gng }|pd S )Nc                 ó’   — g | ]D}t          |¦  «                             ¦   «         ¯#t          |¦  «                             ¦   «         ‘ŒES r=   )r!   Ústrip©r)   Úones     r   rˆ   z,_normalize_filter_values.<locals>.<listcomp>!  s=   € ÐHÐHÐH sµs¸3±x´x·~²~Ñ7GÔ7GÐH•#�c‘(”(—.’.Ñ"Ô"ÐHÐHÐHr   c                 ó^   — g | ]*}|                      ¦   «         ¯|                      ¦   «         ‘Œ+S r=   )r   r  s     r   rˆ   z,_normalize_filter_values.<locals>.<listcomp>#  s-   € ÐRÐRÐR #ÀcÇiÂiÁkÄkÐR�#—)’)‘+”+ÐRÐRÐRr   z[,|/])r.   r0   r1   rL   r!   ÚreÚsplitr   )r*   rý   r6   s      r   Ú_normalize_filter_valuesr    s·   € Ø€}ØˆtÝ�%�$¥¥sÐ+Ñ,Ô,ð DØHÐH¨eÐHÑHÔHˆˆØ	ð D�* U­CÑ0Ô0ð DØRÐR­¬°(¸EÑ)BÔ)BÐRÑRÔRˆˆå),¨U©¬×)9Ò)9Ñ);Ô);ÐC•#�e‘*”*×"Ò"Ñ$Ô$Ð%Ð%ÀˆØˆ>�TÐr   c                 óN  — | �:t          | t          ¦  «        r't          |                      ¦   «         ¦  «        dk    rd S t          | t          ¦  «        r| S t          | ¦  «                             ¦   «                              ¦   «         }|dv rdS |dv rdS t          d| › �¦  «        ‚)Nr   )Ú1ÚtrueÚyesÚyÚonT)Ú0ÚfalseÚnoÚnÚoffFzInvalid finetune filter value: )r.   r!   r3   r   ry   r"   Ú
ValueError)r*   ro   s     r   Ú_normalize_bool_filterr  )  s§   € Ø€}�
 5­#Ñ.Ô.€}µ3°u·{²{±}´}Ñ3EÔ3EÈÒ3JÐ3JØˆtÝ�%�ÑÔð ØˆÝˆu‰:Œ:×ÒÑÔ×&Ò&Ñ(Ô(€DØÐ.Ð.Ð.ØˆtØÐ/Ð/Ð/ØˆuÝ
Ð>°uÐ>Ð>Ñ
?Ô
?Ð?r   c                 ó:  ‡‡— t          ‰¦  «        Š‰€dS t          | t          t          t          f¦  «        r'd„ | D ¦   «         Št          ˆfd„‰D ¦   «         ¦  «        S t          | pd¦  «                             ¦   «         Št          ˆfd„‰D ¦   «         ¦  «        S )NTc                 óP   — g | ]#}t          |¦  «                             ¦   «         ‘Œ$S r=   )r!   r"   r  s     r   rˆ   z,_metadata_matches_filter.<locals>.<listcomp>;  s*   € Ð7Ð7Ð7¨#•#�c‘(”(×#Ò#Ñ%Ô%Ð7Ð7Ð7r   c              3   óp   •K  — | ]0}‰D ]+}t          j        ||                     ¦   «         ¦  «        V — Œ,Œ1d S r<   ©ÚfnmatchÚfnmatchcaser"   )r)   r  ÚpatternÚexpecteds      €r   r+   z+_metadata_matches_filter.<locals>.<genexpr><  sK   øè è € ÐjÐjÀCÐaiÐjÐjÐV]•7Ô& s¨G×,<Ò,<Ñ,>Ô,>Ñ?Ô?ÐjÐjÐjÐjÐjÐjÐjr   r   c              3   óf   •K  — | ]+}t          j        ‰|                     ¦   «         ¦  «        V — Œ,d S r<   r  )r)   r  Úactuals     €r   r+   z+_metadata_matches_filter.<locals>.<genexpr>>  s<   øè è € ÐWÐWÀ7�wÔ" 6¨7×+;Ò+;Ñ+=Ô+=Ñ>Ô>ÐWÐWÐWÐWÐWÐWr   )r  r.   r0   r1   rL   r?   r!   r"   )r*   r  r  s    `@r   Ú_metadata_matches_filterr  6  s¬   øø€ Ý'¨Ñ1Ô1€HØÐØˆtÝ�%�$¥¥sÐ+Ñ,Ô,ð kØ7Ð7°Ð7Ñ7Ô7ˆÝÐjÐjÐjÐjÈ6ÐjÑjÔjÑjÔjÐjÝ��˜"ÑÔ×&Ò&Ñ(Ô(€FÝÐWÐWÐWÐWÈhÐWÑWÔWÑWÔWÐWr   )rí   r   rï   rì   rð   r|   c                ó:  — t          |¦  «        }t          |¦  «        }t          |¦  «        }	t          |d¬¦  «        }
g }|                      ¦   «         D �]D\  }}|                     t          i ¦  «        }t          ||
¦  «        sŒ2t          |                     d¦  «        |¦  «        s$t          |                     d¦  «        |¦  «        sŒyt          |                     d¦  «        |¦  «        sŒ�|�(t          |                     dd¦  «        ¦  «        |k    rŒÇ|�%t          |                     dg ¦  «        |¦  «        sŒî|	�&t          |                     dg ¦  «        |	¦  «        s�Œt          j        |¦  «        }||d	<   | 	                    |¦  «         �ŒF|S )
NFrü   rí   rî   r   rï   rð   r|   rì   )
r  r  rM   r   rú   r  ry   rõ   rö   r2   )Ú
models_defrí   r   rï   rì   rð   r|   Úfinetune_filterÚrequested_outputsÚrequested_inputsÚrequested_model_typesÚrecordsÚone_model_typer   r   Úrecords                   r   Úlist_model_defsr(  A  s»  € Ý,¨XÑ6Ô6€OÝ0°Ñ=Ô=ÐÝ/°Ñ7Ô7ÐÝ4°ZÈeÐTÑTÔTÐØ€GØ%/×%5Ò%5Ñ%7Ô%7ð ñ Ñ!ˆ˜	Ø—=’=¥¨rÑ2Ô2ˆÝ'¨Ð8MÑNÔNð 	ØÝ'¨¯ª°XÑ(>Ô(>ÀÑGÔGð 	ÕPhÐiq×iuÒiuð  wEñ  jFô  jFð  HNñ  QOô  QOð 	ØÝ'¨¯ªÐ5FÑ(GÔ(GÈÑYÔYð 	ØØÐ&­4°·²¸ZÈÑ0OÔ0OÑ+PÔ+PÐTcÒ+cÐ+cØØÐ(Õ1IÈ(Ï,Ê,ÐWdÐfhÑJiÔJiÐk|Ñ1}Ô1}Ð(ØØÐ'Õ0HÈÏÊÐV^Ð`bÑIcÔIcÐeuÑ0vÔ0vÐ'ÙÝ”˜yÑ)Ô)ˆØ-ˆˆ|ÑØ�Š�vÑÔÐÑØ€Nr   )F)rõ   r  r   r  Úhelp_sharedr   rú   rã   r   r$   r8   rA   rP   r\   ri   rm   r   rz   r®   rÑ   rå   rû   r  r  r  r(  r=   r   r   ú<module>r*     s¾  ðà €€€Ø €€€Ø 	€	€	€	Ø 	€	€	€	Ø /Ð /Ð /Ð /Ð /Ð /ð €ð 	Ø	Ø	Ø	Ø	%Ø	-ðð Ð ð(ð (ð (ð (ðmð mð mðð ð ð cð cð cðð ð ð8ð ð ðð ð ðð ð ð'ð 'ð 'ð.2ð 2ð 2ð(ð (ð (ðVð ð ðBð ð ð@ð ð ð: 59ð 	ð 	ð 	ð 	ð 	ð
@ð 
@ð 
@ðXð Xð Xð +/ÀÈtÐ`dÐrvð  @Dð ð ð ð ð ð ð r   