§
    ’Þj  ã                   ó`   — d dl Z d dlmZ ddlmZmZmZmZmZ dZ	dZ
dZdZ G d	„ d
¦  «        ZdS )é    N)Úbuild_hf_urlé   )ÚEVOKE_DEEPY_INFOSÚEVOKE_DEEPY_PROMPT_INFOSÚEVOKE_ENHANCER_INSTRUCTIONSÚEVOKE_INFOSÚEVOKE_PROMPT_INFOSÚevoke_turbozumt5-xxlé$   Fc            
       ó@  — e Zd Ze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ej        ej        d
d
dddf	d„¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         ZdS )Úfamily_handlerc                  ó   — t           gS ©N)ÚARCHITECTURE© ó    úBC:\Users\johnn\OneDrive\Documents\AI\models\evoke\evoke_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 )NÚevoker   r   r   r   Úquery_model_familyz!family_handler.query_model_family   s   € àˆwr   c                  ó
   — ddiS )Nr   )é5   ÚEVOKEr   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                 ón  — i dd“dd“dd“dd“dd	“d
d“dddgg“dt            “dt          “dt          “dt          dz  “dd“dd“dd“dd“dd“dd“ddd d!d dd"gd#œ| gt          t          d$t          d%¦  «        t          d$t          d&¦  «        gt          t
          t          t          d'gd'd(id)d*œt          d+œ
¥S ),NÚdevice_explicitTÚworld_outputsÚdtypeÚbf16ÚacceleratedÚnativeÚfpsé   Úvae_block_sizeé@   Úresolutionsz640x384 (5:3)Ú640x384Úlock_resolutionsÚframes_minimumÚframes_stepsÚframes_selection_maximumé(   Úsliding_windowFÚimage_prompt_types_allowedÚTSÚguidance_max_phasesr   Úno_negative_promptÚinference_stepsÚtiny_vae_architectureÚt2vÚcustom_guideÚWorldz2Continue a World (.wgpworld of the Worlds gallery)Úfilez	.wgpworld)ÚidÚnameÚlabelÚtypeÚdefaultÚrequiredÚ
file_typesúDeepBeepMeep/Wan2.1z'models_t5_umt5-xxl-enc-bf16.safetensorsz.models_t5_umt5-xxl-enc-quanto_int8.safetensorsÚTIz#An EVOKE World from the Start ImageÚ )Ú	selectionÚlabelsrB   )
r;   Úprofiles_dirÚtext_encoder_folderÚtext_encoder_URLsÚinfosÚprompt_infosÚdeepy_infosÚdeepy_prompt_infosÚprompt_enhancer_defÚ"video_prompt_enhancer_instructions)	ÚANY_IMAGE_RATIOÚCHUNK_FRAMESÚTEXT_ENCODER_FOLDERr   r   r	   r   r   r   )r   Ú	model_defs     r   Úquery_model_defzfamily_handler.query_model_def%   s�  € ð 
Ø˜tð 
à˜Tð 
ð �Vð 
ð ˜8ð	 
ð
 �2ð 
ð ˜bð 
ð ˜_¨iÐ8Ð9ð 
ð ¥OÐ 3ð 
ð �lð 
ð �Lð 
ð '­°rÑ(9ð 
ð ˜eð 
ð )¨$ð 
ð " 1ð 
ð ! $ð 
ð  ˜uð! 
ð" $ Uð# 
ð$ $2¸7ð  NBð  LRð  _cØ).¸{¸mðMð Mà,Ð-Ý#6åÐ2Õ4GÐIrÑsÔsÝÐ2Õ4GÐIyÑzÔzð"õ !Ý.Ý,Ý":Ø26°ÀDÐJoÐCpÐ}ð  $Að  $AÝ2Mð? 
ð  
ð  
ð  	
r   c                 ó$   — ddl m}  ||¦  «        S )Nr   )Úexplorer_menus)ÚexplorationrY   )r   ÚmanifestÚmemoryrY   s       r   Úquery_world_explorerz#family_handler.query_world_explorerI   s$   € à/Ð/Ð/Ð/Ð/Ð/Øˆ~˜hÑ'Ô'Ð'r   Nc                 ó8   — dt           dgg d¢dggdœddgdggdœgS )	NrE   rG   )zspecial_tokens_map.jsonzspiece.modelztokenizer.jsonztokenizer_config.jsonzWan2.1_VAE.safetensors)ÚrepoIdÚsourceFolderListÚfileListzDeepBeepMeep/Worldsr   zvigeo1_1.safetensors)rU   )ÚcomputeListr   rV   s      r   Úquery_model_filesz family_handler.query_model_filesN   sY   € ð -ÕCVÐXZÐB[ØpÐpÐpð  tLð  sMð  NðOð Oà,À7À)Ð[qÐZrÐYsÐtÐtð
ð 	
r   Féÿÿÿÿc                 ó²   — ddl m}  || ||||	|¬¦  «        }|j        |j        j        |j        j        |j        dœ}|dgddgdgdœd	œ}|d
v rddg|d<   ||fS )Nr   )Úmodel_factory)Úsave_quantizedÚ	VAE_dtype)ÚtransformerÚtext_encoderÚvaeÚvigeork   ri   rj   )ri   rk   rj   )ÚpipeÚcoTenantsMap)é   é   rl   ÚpinnedMemory)Ú
evoke_mainrf   ri   rj   Úmodelrk   rl   )Úmodel_filenameÚ
model_typer   rV   ÚquantizeTransformerÚtext_encoder_quantizationr$   rh   Úmixed_precision_transformerrg   Úsubmodel_no_listÚtext_encoder_filenameÚprofileÚkwargsrf   rs   rm   Úoffload_kwargss                     r   Ú
load_modelzfamily_handler.load_modelV   sÔ   € ð 	.Ð-Ð-Ð-Ð-Ð-à�˜n¨j¸/ÐK`Ðqð  LUð  Vñ  Vô  VˆØ$Ô0À%ÔBTÔBZÐchÔclÔcrð  ~Cô  ~Ið  Jð  Jˆð #'ÈÈÐYfÐhvÐXwð  KPð  JQð  9Rð  9Rð  Sð  SˆØ�fÐÐØ.;¸WÐ-EˆN˜>Ñ*Ø�nÐ$Ð$r   c                 ó&   — ddl m}  |dd¦  «        S )Nr   )Úget_rgb_factorsÚwanr:   )Úshared.RGB_factorsr€   )r   r€   s     r   r€   zfamily_handler.get_rgb_factorsd   s&   € à6Ð6Ð6Ð6Ð6Ð6Øˆ˜u eÑ,Ô,Ð,r   c                 óp   — |d         �d S d|d         vrdS ddl m}  ||d         ¦  «        d         sdS d S )	Nr;   ÚSÚimage_prompt_typez}A new EVOKE world starts from a Start Image (Start with Image); without one, choose a world of the Worlds gallery to continuer   )Úsplit_instructionsÚpromptzVA new EVOKE world needs a description of the place before its exploration instructions)Úshared.utils.world_instructionsr†   )r   rV   Úinputsr†   s       r   Úvalidate_generative_settingsz+family_handler.validate_generative_settingsi   sj   € à�.Ô!Ð-Ø�4Ø�fÐ0Ô1Ð1Ð1ð Sð  SØFÐFÐFÐFÐFÐFØ!Ð! &¨Ô"2Ñ3Ô3°AÔ6ð 	lØkÐkØˆtr   c           	      óN   — |                      ddt          dz  ddddœ¦  «         d S )Nro   g      ð?rp   r-   r„   ÚFG)Únum_inference_stepsÚguidance_scaleÚvideo_lengthÚ
resolutionr…   Úmulti_prompts_gen_type)ÚupdaterT   )r   rV   Úui_defaultss      r   Úupdate_default_settingsz&family_handler.update_default_settingst   sE   € à×ÒØ#$Ø!Ý(¨1Ñ,Ø#Ø!$Ø&*ð
ð 
ñ 	ô 	ð 	ð 	ð 	r   r   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r   r   r    rW   r]   rc   ÚtorchÚbfloat16Úfloat32r~   r€   rŠ   r”   r   r   r   r   r      s”  € € € € € Øðð ñ „\ðð ðð ñ „\ðð ðð ñ „\ðð ð(ð (ñ „\ð(ð ðð ñ „\ðð ð!
ð !
ñ „\ð!
ðF ð(ð (ñ „\ð(ð ð
ð 
ð 
ñ „\ð
ð Ø_dð  AEð  MRô  M[Ø"œ]ÈÐ^cÐvzð  SWð  acð%ð %ð %ñ „\ð%ð ð-ð -ñ „\ð-ð ðð ñ „\ðð ðð ñ „\ðð ð r   r   )r™   Úshared.utils.hfr   Útextsr   r   r   r   r	   r   rU   rT   rS   r   r   r   r   ú<module>rž      s�   ðà €€€à (Ð (Ð (Ð (Ð (Ð (Ø |Ð |Ð |Ð |Ð |Ð |Ð |Ð |Ð |Ð |Ð |Ð |Ð |Ð |à€Ø Ð Ø€ð €ðmð mð mð mð mñ mô mð mð mð mr   