§
    Û×jÁ  ã                   óP   — d dl mZmZmZmZmZ d dlmZ dZdZ	 G d„ d¦  «        Z
dS )é   )ÚASSET_NAME_INSTRUCTIONSÚTRIPOSPLAT_DEEPY_INFOSÚTRIPOSPLAT_DEEPY_PROMPT_INFOSÚTRIPOSPLAT_INFOSÚTRIPOSPLAT_PROMPT_INFOS)ÚCUSTOM_SETTINGSÚ
triposplatzDeepBeepMeep/3Dc                   óØ   — 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© ó    úLC:\Users\johnn\OneDrive\Documents\AI\models\triposplat\triposplat_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Ö  Ú
TripoSplatr   r   r   r   Úquery_family_infosz!family_handler.query_family_infos   s   € àÐ2Ð3Ð3r   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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“d| g“d„ t           D ¦   «         dgddiddœt          dt          t          t          t
          d g d!¢d"d#œgd$œ	¥S )%NÚimage_outputsTÚmesh_outputsÚdevice_explicitÚdtypeÚfp16Úimage_ref_choices)zObject ImageÚIr$   )ÚchoicesÚletters_filterÚdefaultÚone_image_ref_neededÚone_image_ref_onlyÚimage_refs_keep_alphaÚprompt_descriptionz8Asset Name / Description (optional, used for file names)Úprompt_optionalÚno_negative_promptÚguidance_max_phasesr   Úembedded_guidanceFÚNAGÚinference_stepsÚ
flow_shiftÚprofiles_dirc                 ó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   ÚTIz%An Asset Name from the {image_inputs}Ú )Ú	selectionÚlabelsr'   é    zimage to 3D Gaussian splat)z3D Gaussian splattingÚ3DGSÚsplatz	PLY splatzZTurns one picture of an object into a 3D Gaussian splat (PLY) for photographic 3D viewing.)ÚnameÚaliasesÚdescription)	Úcustom_settingsÚprompt_enhancer_defÚ"image_prompt_enhancer_instructionsÚ image_prompt_enhancer_max_tokensÚinfosÚprompt_infosÚdeepy_infosÚdeepy_prompt_infosÚspecialities)r   r   r   r   r   r   )r   Ú	model_defs     r   Úquery_model_defzfamily_handler.query_model_def   sm  € ð
Ø˜Tð
à˜Dð
ð ˜tð
ð �Vð	
ð
  Ð.CÐ-DÐX[ÐhkÐ!lÐ!lð
ð # Dð
ð ! $ð
ð $ Tð
ð !Ð"\ð
ð ˜tð
ð ! $ð
ð " 1ð
ð   ð
ð �5ð
ð ˜tð
ð  ˜$ð!
ð" ˜_Ð-ð#
ð$  GÐFµoÐFÑFÔFØ26°ÀDÐJqÐCrð  @Bð  $Cð  $CÝ2IØ02Ý%Ý3Ý1Ý"?Ø&Bð  PGð  PGð  PGð .JðKð Kð Lð5
ð 
ð 
ð 	
r   Nc                 ó8   — t           ddgdgddggdœddgd	ggdœgS )
Nr	   Údinov3_vith16plusz'triposplat_vae_decoder_fp16.safetensorsz+dinov3_vith16plus_lvd1689m_bf16.safetensorszLICENSE_DINOv3.md)ÚrepoIdÚsourceFolderListÚfileListzDeepBeepMeep/Flux2r:   zflux2_vae.safetensors)ÚREPO)ÚcomputeListr   rL   s      r   Úquery_model_filesz family_handler.query_model_files>   sX   € õ °,Ð@SÐ1TØDÐEÐHuð  xKð  HLð  MðNð Nà+À"ÀÐUlÐTmÐSnÐoÐoð
ð 	
r   Fc                 óX   — ddl m}  || |||¬¦  «        }||                     ¦   «         fS )Nr   )Úmodel_factory)Úsave_quantized)Útriposplat_mainrW   Úpipe)Úmodel_filenameÚ
model_typer   rL   rX   ÚkwargsrW   Úmodels           r   Ú
load_modelzfamily_handler.load_modelF   s@   € à2Ð2Ð2Ð2Ð2Ð2à�˜n¨j¸/ÐZhÐiÑiÔiˆØ�e—j’j‘l”lÐ"Ð"r   c                 óB   — |                      dddddddddœ¦  «         d S )Nr   r$   é   g      @r:   )Ú
image_modeÚvideo_prompt_typeÚnum_inference_stepsÚguidance_scaler2   Ú
batch_sizeÚremove_background_images_refÚprompt_enhancer)Úupdate)r   rL   Úui_defaultss      r   Úupdate_default_settingsz&family_handler.update_default_settingsM   sF   € à×ÒØØ!$Ø#%Ø!ØØØ,-Ø!ð	
ð 	
ñ 		ô 		ð 		ð 		ð 		r   r   )F)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r   r   r   rM   rU   r_   rk   r   r   r   r   r   	   s  € € € € € Øðð ñ „\ðð ðð ñ „\ðð ðð ñ „\ðð ð4ð 4ñ „\ð4ð ðð ñ „\ðð ð
ð 
ñ „\ð
ð> ð
ð 
ð 
ñ „\ð
ð ð#ð #ð #ñ „\ð#ð ð
ð 
ñ „\ð
ð 
ð 
r   r   N)Úpromptsr   r   r   r   r   Úsettingsr   r   rS   r   r   r   r   ú<module>rr      s±   ðð Oð  Oð  Oð  Oð  Oð  Oð  Oð  Oð  Oð  Oð  Oð  Oð  Oð  OØ %Ð %Ð %Ð %Ð %Ð %à€Ø€ðOð Oð Oð Oð Oñ Oô Oð Oð Oð Or   