§
    Û×j9  ã                   óP   — d dl mZmZmZmZ d dlmZmZ dZdZ	 G d„ d¦  «        Z
dS )é   )ÚTRELLIS2_DEEPY_INFOSÚTRELLIS2_DEEPY_PROMPT_INFOSÚTRELLIS2_INFOSÚTRELLIS2_PROMPT_INFOS)ÚCUSTOM_SETTINGSÚMODEL_MODESÚtrellis2z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© ó    úHC:\Users\johnn\OneDrive\Documents\AI\models\trellis2\trellis2_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   € àˆzr   c                  ó
   — ddiS )Nr	   )iÌ  z	TRELLIS.2r   r   r   r   Úquery_family_infosz!family_handler.query_family_infos   s   € àÐ/Ð0Ð0r   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i t           ¥dt          t           d         ¦  «        i¥“d„ t          D ¦   «         g i ddœ| gd g d!¢d"d#œd$g d%¢d&d#œgt          t          t
          t          d'œ¥S )(NÚimage_outputsTÚmesh_outputsÚdevice_explicitÚdtypeÚbf16Úimage_ref_choices)zObject ImageÚIr#   )ÚchoicesÚletters_filterÚdefaultÚone_image_ref_neededÚone_image_ref_onlyÚimage_refs_keep_alphaÚfit_into_canvas_image_refsé    Úprompt_descriptionz8Asset Name / Description (optional, used for file names)Úprompt_optionalÚno_negative_promptÚembedded_guidanceFÚinference_stepsÚguidance_max_phasesr   Úalt_guidancezTexture GuidanceÚmodel_modesr$   c                 ó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   Ú )Ú	selectionÚlabelsr&   zimage to 3D)z3D meshÚGLBz3D assetz.Turns one picture of an object into a 3D mesh.)ÚnameÚaliasesÚdescriptionzPBR textured 3D asset)ztextured 3D modelztextured GLBzimage to textured 3Dz
game asseta  Turns one picture of an object into a GLB with baked PBR textures (base colour, metallic, roughness) in one pass, keeping thin and intricate structures. Prefer Hunyuan3D to texture an existing mesh, TripoSG for an untextured shape and TripoSplat for a Gaussian splat.)Úcustom_settingsÚprompt_enhancer_defÚprofiles_dirÚspecialitiesÚinfosÚprompt_infosÚdeepy_infosÚdeepy_prompt_infos)r   Úlistr   r   r   r   r   )r   Ú	model_defs     r   Úquery_model_defzfamily_handler.query_model_def   s–  € ð
Ø˜Tð
à˜Dð
ð ˜tð
ð �Vð	
ð
  Ð.CÐ-DÐX[ÐhkÐ!lÐ!lð
ð # Dð
ð ! $ð
ð $ Tð
ð )¨!ð
ð !Ð"\ð
ð ˜tð
ð ! $ð
ð   ð
ð ˜tð
ð " 1ð
ð  Ð.ð!
ð" ÐS�kÐS¨9µd½;ÀyÔ;QÑ6RÔ6RÐSÐSð#
ð$  GÐFµoÐFÑFÔFØ13¸rÈbÐ#QÐ#QØ,Ð-Ø&3Ð@^Ð@^Ð@^ð  p`ð  að  aØ&=ð  KVð  KVð  KVð.Zð[ð [ð\õ
 $Ý1Ý/Ý"=ð;
ð 
ð 
ð 	
r   Nc                 ó*   — t           ddgg d¢ddggdœgS )Nr	   Údinov3_vitl16)z$trellis2_ss_decoder_fp16.safetensorsz'trellis2_shape_decoder_fp16.safetensorsz)trellis2_texture_decoder_fp16.safetensorsz"dinov3_vitl16_lvd1689m.safetensorszLICENSE_DINOv3.md)ÚrepoIdÚsourceFolderListÚfileList)ÚREPO)ÚcomputeListr   rI   s      r   Úquery_model_filesz family_handler.query_model_files@   sD   € õ Ø!+¨_Ð =ð Zð  Zð  ZØ>Ð@SÐTðVð
ð 
ð ð 	r   Fc                 óX   — ddl m}  || |||¬¦  «        }||                     ¦   «         fS )Nr   )Úmodel_factory)Úsave_quantized)Útrellis2_mainrT   Úpipe)Úmodel_filenameÚ
model_typer   rI   rU   ÚkwargsrT   Úmodels           r   Ú
load_modelzfamily_handler.load_modelI   s@   € à0Ð0Ð0Ð0Ð0Ð0à�˜n¨j¸/ÐZhÐiÑiÔiˆØ�e—j’j‘l”lÐ"Ð"r   c                 óX   — |                      dddddt          d         dddœ¦  «         d S )Nr   r#   é   g      @g      ð?r&   )Ú
image_modeÚvideo_prompt_typeÚnum_inference_stepsÚguidance_scaleÚalt_guidance_scaleÚ
model_modeÚ
batch_sizeÚremove_background_images_ref)Úupdater   )r   rI   Úui_defaultss      r   Úupdate_default_settingsz&family_handler.update_default_settingsP   sK   € à×ÒØØ!$Ø#%Ø!Ø"%Ý% iÔ0ØØ,-ð	
ð 	
ñ 		ô 		ð 		ð 		ð 		r   r   )F)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r   r   r   rJ   rR   r\   ri   r   r   r   r   r   	   s  € € € € € Øðð ñ „\ðð ðð ñ „\ðð ðð ñ „\ðð ð1ð 1ñ „\ð1ð ðð ñ „\ðð ð
ð 
ñ „\ð
ðB ðð ð ñ „\ðð ð#ð #ð #ñ „\ð#ð ð
ð 
ñ „\ð
ð 
ð 
r   r   N)Úpromptsr   r   r   r   Úsettingsr   r   r   rP   r   r   r   r   ú<module>rp      s‡   ðà mÐ mÐ mÐ mÐ mÐ mÐ mÐ mÐ mÐ mÐ mÐ mØ 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2à€Ø€ðRð Rð Rð Rð Rñ Rô Rð Rð Rð Rr   