§
     ’ÞjÕ  ã                   óP   — d dl Z ddlmZmZmZmZ dddœZd„ Z G d„ d	¦  «        ZdS )
é    Né   )ÚWAYPOINT_DEEPY_INFOSÚWAYPOINT_DEEPY_PROMPT_INFOSÚWAYPOINT_INFOSÚWAYPOINT_PROMPT_INFOS)é   i   )é   r   )Úwaypoint_1_5Úwaypoint_1_5_360pc                 ó0   — t           |          \  }}|› d|› �S )NÚx)ÚARCHITECTURES)Úbase_model_typeÚheightÚwidths      úHC:\Users\johnn\OneDrive\Documents\AI\models\waypoint\waypoint_handler.pyÚ_resolutionr   	   s$   € Ý! /Ô2�M€FˆEØÐÐ�fÐÐÐó    c            
       ó@  — 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          t          ¦  «        S ©N)Úlistr   © r   r   Úquery_supported_typesz$family_handler.query_supported_types   s   € å•MÑ"Ô"Ð"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Úwaypointr   r   r   r   Úquery_model_familyz!family_handler.query_model_family   s   € àˆzr   c                  ó
   — ddiS )Nr   )é6   ÚWaypointr   r   r   r   Úquery_family_infosz!family_handler.query_family_infos   s   € àÐ,Ð-Ð-r   c                 ó   — | S r   r   )r   s    r   Úget_lora_dirzfamily_handler.get_lora_dir   s   € àÐr   c                 ó  — t          | ¦  «        }i dd“dd“ddg“dd“dd	“d
| dk    rdnd“ddg d¢ddœg“dd“dd“d|› d�|gg“dd“dd“dd“dd“dd “d!d"“d#d$“dd%d d&d'd(d)d*d d d+gd,œ| gt          t          t          t          d-œ
¥S ).NÚdevice_explicitTÚworld_outputsÚcuda_graphsÚtransformerÚdtypeÚbf16ÚacceleratedÚnativeÚsizer   ÚlighterÚlargeÚspecialitieszreal-time game worlds)z
video gameÚinteractivez	real timezUplayable 60 fps game worlds steered with the keyboard, generated faster than playback)ÚnameÚaliasesÚdescriptionÚfpsé<   Úvae_block_sizeé    Úresolutionsz (2:1)Úlock_resolutionsÚframes_minimumé   Úframes_stepsÚframes_selection_maximumi   Úsliding_windowFÚimage_prompt_types_allowedÚTSÚguidance_max_phasesr   zªMoves, one per line, e.g. [Forward 2s], [Forward+Jump], [Look_Left 1s] (no description: Waypoint reads no text; without moves the world stands still for the length below)Úhunyuan_1_5_t2vÚcustom_guideÚWorldz2Continue a World (.wgpworld of the Worlds gallery)Úfilez	.wgpworld)Úidr5   ÚlabelÚtypeÚdefaultÚrequiredÚ
file_types)
Úno_negative_promptÚprompt_descriptionÚinference_stepsÚtiny_vae_architecturerG   Úprofiles_dirÚinfosÚprompt_infosÚdeepy_infosÚdeepy_prompt_infos)r   r   r   r   r   )r   Ú	model_defÚ
resolutions      r   Úquery_model_defzfamily_handler.query_model_def#   s„  € å  Ñ1Ô1ˆ
ð
Ø˜tð
à˜Tð
ð ˜M˜?ð
ð �Vð	
ð
 ˜8ð
ð  Ð4GÒ!GÐ!G�I�IÈWð
ð Ð&=ÐJtÐJtÐJtð .EðFð Fð Gð
ð �2ð
ð ˜bð
ð  
Ð2Ð2Ð2°JÐ?Ð@ð
ð  ð
ð ˜að
ð ˜Að
ð '¨ð
ð  ˜eð!
ð" )¨$ð#
ð$ " 1ð%
ð& #'ð #OØ$Ø%6Ø#1¸7ð  NBð  LRð  _cØ).¸{¸mðMð Mà,Ð-Ý#Ý1Ý/Ý"=ð=
ð 
ð 
ð 	
r   c                 ó$   — ddl m}  ||¦  «        S )Nr   )Úexplorer_menus)Úexplorationr]   )r   ÚmanifestÚmemoryr]   s       r   Úquery_world_explorerz#family_handler.query_world_explorerG   s$   € à/Ð/Ð/Ð/Ð/Ð/Øˆ~˜hÑ'Ô'Ð'r   Nc                 ó   — ddgdggdœgS )NzDeepBeepMeep/HunyuanVideo1.5Úpreview_decodersztaehv1_5.safetensors)ÚrepoIdÚsourceFolderListÚfileListr   )ÚcomputeListr   rY   s      r   Úquery_model_filesz family_handler.query_model_filesL   s8   € à9ÐPbÐOcð  tJð  sKð  rLð  Mð  Mð  Nð  	Nr   Féÿÿÿÿc                 óˆ   — ddl m} t          |         \  }} || |||	||¬¦  «        }||j        |j        dœdgdgdœddidœfS )	Nr   )Úmodel_factory)Úsave_quantizedr   r   )r+   Úvaerm   r+   r   )ÚpipeÚcoTenantsMapÚbudgets)Úwaypoint_mainrk   r   r+   rm   )Úmodel_filenameÚ
model_typer   rY   ÚquantizeTransformerÚtext_encoder_quantizationr,   Ú	VAE_dtypeÚmixed_precision_transformerrl   Úsubmodel_no_listÚtext_encoder_filenameÚprofileÚkwargsrk   r   r   Úmodels                     r   Ú
load_modelzfamily_handler.load_modelP   s¢   € ð 	1Ð0Ð0Ð0Ð0Ð0å% oÔ6‰ˆ�Ø�˜n¨j¸/ÐZhÐqwð  @Eð  Fñ  Fô  Fˆð ¨uÔ/@ÈÌÐSÐSÐv{Ðu|ð  GTð  FUð  fVð  fVØ#0°!Ð"4ð6ð 6ð 6ð 	6r   c                 ó(   — ddl m}  |dd¬¦  «        S )Nr   )Úget_rgb_factorsÚhunyuanz
hunyuan1.5)Ú
sub_family)Úshared.RGB_factorsr   )r   r   s     r   r   zfamily_handler.get_rgb_factors]   s)   € à6Ð6Ð6Ð6Ð6Ð6Øˆ˜y°\ÐBÑBÔBÐBr   c                 óº   — |d         €d|d         vrdS ddl m} ddlm} 	  ||d	         |¦  «         n&# t          $ r}t          |¦  «        cY d }~S d }~ww xY wd S )
NrG   ÚSÚimage_prompt_typez€A new Waypoint world starts from a Start Image (Start with Image); without one, choose a world of the Worlds gallery to continuer   )Úsplit_instructionsr   )ÚEXPLORER_CONTROLSÚprompt)Úshared.utils.world_instructionsr†   Úcontrolsr‡   Ú
ValueErrorÚstr)r   rY   Úinputsr†   r‡   Úerrors         r   Úvalidate_generative_settingsz+family_handler.validate_generative_settingsb   s­   € à�.Ô!Ð)¨c¸Ð@SÔ9TÐ.TÐ.Tð Vð  VØFÐFÐFÐFÐFÐFØ/Ð/Ð/Ð/Ð/Ð/ð	ØÐ˜v hÔ/Ð1BÑCÔCÐCÐCøÝð 	ð 	ð 	Ý�u‘:”:ÐÐÐÐÐÐøøøøð	øøøàˆts   ¢5 µ
A¿AÁAÁAc           	      óX   — |                      dddt          | ¦  «        dddœ¦  «         d S )Nr?   g      ð?éð   r„   ÚFG)Únum_inference_stepsÚguidance_scaleÚvideo_lengthrZ   r…   Úmulti_prompts_gen_type)Úupdater   )r   rY   Úui_defaultss      r   Úupdate_default_settingsz&family_handler.update_default_settingsn   sH   € à×ÒØ#$Ø!ØÝ% oÑ6Ô6Ø!$Ø&*ð
ð 
ñ 	ô 	ð 	ð 	ð 	r   r   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r    r$   r&   r[   ra   rh   ÚtorchÚbfloat16Úfloat32r}   r   r�   r™   r   r   r   r   r      s›  € € € € € Øð#ð #ñ „\ð#ð ðð ñ „\ðð ðð ñ „\ðð ð.ð .ñ „\ð.ð ðð ñ „\ðð ð!
ð !
ñ „\ð!
ðF ð(ð (ñ „\ð(ð ðNð Nð Nñ „\ðNð Ø_dð  AEð  MRô  M[Ø"œ]ÈÐ^cÐvzð  SWð  acð
6ð 
6ð 
6ñ „\ð
6ð ðCð Cñ „\ðCð ð	ð 	ñ „\ð	ð ðð ñ „\ðð ð r   r   )	rž   Útextsr   r   r   r   r   r   r   r   r   r   ú<module>r¢      sŠ   ðà €€€à kÐ kÐ kÐ kÐ kÐ kÐ kÐ kÐ kÐ kÐ kÐ kà!,À:ÐNÐN€ðð ð ð
ið ið ið ið iñ iô ið ið ið ir   