§
    ÎÐj†  ã                   óz   — d dl mZ ddlmZmZmZmZmZ ddlm	Z	 dZ
dZdZd\  ZZg d¢Zg d	¢Z G d
„ d¦  «        ZdS )é    )Úbuild_hf_urlé   )ÚDEEPY_INFOSÚDEEPY_PROMPT_INFOSÚINFOSÚMOTION_ENHANCERÚPROMPT_INFOS)ÚCUSTOM_SETTINGSÚanimate_any_meshzDeepBeepMeep/3D)zDeepBeepMeep/HunyuanVideoÚclip_vit_large_patch14)ztext_config.jsonz
merges.txtzspecial_tokens_map.jsonztokenizer.jsonztokenizer_config.jsonz
vocab.json)z.glbz.objz.plyz.stlc                   óØ   — e Zd Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Z	edd„¦   «         Z
edd
„¦   «         Zed„ ¦   «         ZdS )Úfamily_handlerc                  ó   — t           gS ©N)ÚARCHITECTURE© ó    úXC:\Users\johnn\OneDrive\Documents\AI\models\animate_any_mesh\animate_any_mesh_handler.pyÚquery_supported_typesz$family_handler.query_supported_types   s
   € åˆ~Ðr   c                  ó
   — i i fS r   r   r   r   r   Úquery_family_mapsz family_handler.query_family_maps   s   € à�2ˆvˆr   c                  ó   — dS )Nr   r   r   r   r   Úquery_model_familyz!family_handler.query_model_family   s   € à!Ð!r   c                  ó
   — ddiS )Nr   )ià  ÚAnimateAnyMeshr   r   r   r   Úquery_family_infosz!family_handler.query_family_infos   s   € à"Ð$<Ð=Ð=r   c                 ó   — | S r   r   )Úbase_model_types    r   Úget_lora_dirzfamily_handler.get_lora_dir!   s   € àÐr   c           
      ó@  — i dd“dd“dd“dd“dd“dd	“d
d“dd“dddddd dt           dœ“dd„ t          D ¦   «         “dt          t          t          d¦  «        g“dt          “ddgddiddœ“dt
          “dd“d| g“d t          “t          t          t          d!g d"¢d#d$œgd%œ¥S )&NÚimage_outputsTÚmesh_outputsÚdevice_explicitÚno_negative_promptÚembedded_guidanceFÚguidance_max_phasesr   Úinference_stepsÚprompt_descriptionzGMotion Description (English), e.g. The dragon flaps its wings and fliesÚcustom_guideÚMeshz
Input MeshÚfile)ÚidÚnameÚlabelÚtypeÚdefaultÚrequiredÚ
file_typesÚcustom_settingsc                 ó6   — g | ]}|                      ¦   «         ‘ŒS r   )Úcopy)Ú.0Úones     r   ú
<listcomp>z2family_handler.query_model_def.<locals>.<listcomp>2   s    € ÐFÐFÐF¨s §¢¡
¤
ÐFÐFÐFr   Útext_encoder_URLszmodel.safetensorsÚtext_encoder_folderÚprompt_enhancer_defÚTz3A Motion Description using the existing Text PromptÚ )Ú	selectionÚlabelsr0   Ú!text_prompt_enhancer_instructionsÚtext_prompt_enhancer_max_tokensé@   Úprofiles_dirÚinfoszmesh animation)zanimate a meshzanimated 3DÚ4Dzmorph animationzrigless animationzTAnimates a static 3D mesh (16 frames, no rig) from a text description of its motion.)r-   ÚaliasesÚdescription)Úprompt_infosÚdeepy_infosÚdeepy_prompt_infosÚspecialities)
ÚMESH_FILE_TYPESr
   r   ÚTEXT_ENCODER_REPOÚTEXT_ENCODER_FOLDERr   r   r	   r   r   )r   Ú	model_defs     r   Úquery_model_defzfamily_handler.query_model_def%   s  € ð
Ø˜Tð
à˜Dð
ð ˜tð
ð ! $ð	
ð
   ð
ð " 1ð
ð ˜tð
ð !Ð"kð
ð  >¸6ÈLÐbhÐuyð  HLÝ+:ð<ð <ð
ð ÐFÐFµoÐFÑFÔFð
ð  ¥,Õ/@ÕBUÐWjÑ"kÔ"kÐ!lð
ð "Õ#6ð
ð "°#°À3ÐH}ÐB~ð  LNð  $Oð  $Oð
ð 0µð
ð  .¨rð!
ð" ˜_Ð-ð#
ð$ •Uð%
õ& )Ý&Ý"4Ø&6ð  DSð  DSð  DSð .DðEð Eð Fð-
ð 
ð 
ð 	
r   Nc                 óV   — t           t          gdggdœt          t          gt          gdœgS )Nz%animate_any_mesh_vae_fp32.safetensors)ÚrepoIdÚsourceFolderListÚfileList)ÚREPOÚPROJECTrM   rN   ÚTOKENIZER_FILES)ÚcomputeListr   rO   s      r   Úquery_model_filesz family_handler.query_model_filesA   s9   € åµg°YÐNuÐMvÐLwÐxÐxÝ,ÕCVÐBWÕfuÐevÐwÐwðyð 	yr   Fc                 óZ   — ddl m}  || ||||¬¦  «        }||                     ¦   «         fS )Nr   )Úmodel_factory)Úsave_quantized)Úanimate_any_mesh_mainr[   Úpipe)	Úmodel_filenameÚ
model_typer   rO   Útext_encoder_filenamer\   Úkwargsr[   Úmodels	            r   Ú
load_modelzfamily_handler.load_modelF   sL   € à8Ð8Ð8Ð8Ð8Ð8à�˜n¨j¸/ÐK`Ðqð  Añ  Aô  AˆØ�e—j’j‘l”lÐ"Ð"r   c                 ó<   — |                      ddddddœ¦  «         d S )Nr   é?   g      @r=   )Ú
image_modeÚnum_inference_stepsÚguidance_scaleÚ
batch_sizeÚprompt_enhancer)Úupdate)r   rO   Úui_defaultss      r   Úupdate_default_settingsz&family_handler.update_default_settingsM   s=   € à×ÒØØ#%Ø!ØØ!ð
ð 
ñ 	ô 	ð 	ð 	ð 	r   r   )F)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r   r   r   rP   rY   rd   rn   r   r   r   r   r      s  € € € € € Øðð ñ „\ðð ðð ñ „\ðð ð"ð "ñ „\ð"ð ð>ð >ñ „\ð>ð ðð ñ „\ðð ð
ð 
ñ „\ð
ð6 ðyð yð yñ „\ðyð ð#ð #ð #ñ „\ð#ð ðð ñ „\ðð ð r   r   N)Úshared.utils.hfr   Úpromptsr   r   r   r   r	   Úsettingsr
   r   rU   rV   rM   rN   rW   rL   r   r   r   r   ú<module>rv      sÉ   ðà (Ð (Ð (Ð (Ð (Ð (à ZÐ ZÐ ZÐ ZÐ ZÐ ZÐ ZÐ ZÐ ZÐ ZÐ ZÐ ZÐ ZÐ ZØ %Ð %Ð %Ð %Ð %Ð %à!€Ø€Ø
€à)^Ñ &Ð Ð&ð Ið  Ið  I€Ø2Ð2Ð2€ðEð Eð Eð Eð Eñ Eô Eð Eð Eð Er   