§
    ÎÐj¢  ã                   ól   — d dl Z d dlmZ d dlmZ ddlmZmZmZm	Z	m
Z
mZ dZdZdZ G d	„ d
¦  «        ZdS )é    N)Úspecialities)Úbuild_hf_urlé   )ÚK6_DEEPY_INFOSÚK6_DEEPY_PROMPT_INFOSÚK6_IMAGE_ENHANCER_INSTRUCTIONSÚK6_INFOSÚK6_PROMPT_INFOSÚK6_TEXT_ENHANCER_INSTRUCTIONSÚk6_pro_distillÚk6_lite_distillzQwen2.5-VL-7B-Instructc            	       ó  — 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fd„¦   «         Zed„ ¦   «         ZdS )Úfamily_handlerc                  ó   — t           t          gS ©N)ÚPRO_DISTILL_ARCHÚLITE_DISTILL_ARCH© ó    úLC:\Users\johnn\OneDrive\Documents\AI\models\kandinsky6\kandinsky6_handler.pyÚquery_supported_typesz$family_handler.query_supported_types   s   € å Õ"3Ð4Ð4r   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Ú
kandinsky6r   r   r   r   Úquery_model_familyz!family_handler.query_model_family   s   € àˆ|r   c                  ó
   — ddiS )Nr   )é3   zKandinsky 6r   r   r   r   Úquery_family_infosz!family_handler.query_family_infos   s   € àÐ1Ð2Ð2r   c                 ó   — | S r   r   )Úbase_model_types    r   Úget_lora_dirzfamily_handler.get_lora_dir   s   € àÐr   c                 ó$   — ddl m}  |d¦  «        S )Nr   )Úget_rgb_factorsÚhunyuan)Úshared.RGB_factorsr$   )r!   r$   s     r   r$   zfamily_handler.get_rgb_factors"   s$   € à6Ð6Ð6Ð6Ð6Ð6àˆ˜yÑ)Ô)Ð)r   c                 ód  — i dd“ddg“dd“dd“d	| t           k    rd
nd“dd“dd“dd“dd“dd“dd“dd“dd“dddi“dd“dd “d!d“d| gt          t          d"t          d#¦  «        t          d"t          d$¦  «        gt          t	          d%¦  «        t
          t          t          dt          t          d&d&d'œ¥S )(NÚdevice_explicitTÚcuda_graphsÚtransformerÚdtypeÚbf16Útiny_vae_architecturer%   ÚsizeÚlargeÚlighterÚacceleratedÚnativeÚreturns_audioÚmultimedia_generationÚfpsé   Úframes_minimumé   Úframes_stepsé   Úframes_selection_maximuméñ   Úsliding_windowFÚsliding_window_defaultsÚwindow_defaultéy   Úimage_prompt_types_allowedÚTSÚguidance_max_phasesr   Úno_negative_promptúDeepBeepMeep/Qwen_imagez'Qwen2.5-VL-7B-Instruct_bf16.safetensorsz3Qwen2.5-VL-7B-Instruct_quanto_bf16_int8.safetensorszaudio-video generationi   )Úinference_stepsÚprofiles_dirÚtext_encoder_folderÚtext_encoder_URLsÚinfosr   Úprompt_infosÚdeepy_infosÚdeepy_prompt_infosÚprompt_enhancer_video_durationÚ!text_prompt_enhancer_instructionsÚ"video_prompt_enhancer_instructionsÚtext_prompt_enhancer_max_tokensÚ video_prompt_enhancer_max_tokens)
r   ÚTEXT_ENCODER_FOLDERr   r	   r   r
   r   r   r   r   )r!   Ú	model_defs     r   Úquery_model_defzfamily_handler.query_model_def(   se  € ð#
Ø˜tð#
à˜M˜?ð#
ð �Vð#
ð $ Yð	#
ð
 ˜Õ2BÒBÐB�G�GÈ	ð#
ð ˜8ð#
ð ˜Tð#
ð $ Tð#
ð �2ð#
ð ˜að#
ð ˜Að#
ð '¨ð#
ð ˜eð#
ð &Ð(8¸#Ð'>ð#
ð )¨$ð#
ð  " 1ð!#
ð" ! $ð##
ð$  $Ø,Ð-Ý#6åÐ6Õ8KÐMvÑwÔwÝÐ6Õ8Kð  NCñ  Dô  Dð"õ Ý(Ð)AÑBÔBÝ+Ý)Ý"7Ø.2Ý1NÝ2PØ/3Ø04ðE#
ð #
ð #
ð #	
r   Nc                 óT   — dt           gg d¢gdœdddgg d¢dd	ggdœd
ddgdgddggdœgS )NrE   )úconfig.jsonú
merges.txtútokenizer_config.jsonú
vocab.json)ÚrepoIdÚsourceFolderListÚfileListzDeepBeepMeep/HunyuanVideoÚclip_vit_large_patch14Ú )ztext_config.jsonrX   zmodel.safetensorszspecial_tokens_map.jsonztokenizer.jsonrY   rZ   z"hunyuan_video_VAE_fp32.safetensorszhunyuan_video_VAE_config.jsonzDeepBeepMeep/Wan2.1ÚmmaudioÚbigvgan_v2_44khz_128band_512xz	v1-44.pthrW   zbigvgan_generator.pt)rS   )ÚcomputeListr!   rT   s      r   Úquery_model_filesz family_handler.query_model_filesO   sŒ   € ð 4Ý%8Ð$9ØaÐaÐaÐbðð ð 6Ø%=¸rÐ$Bð `ð  `ð  `Ø9Ð;ZÐ[ððð ð 0Ø%.Ð0OÐ$PØ)˜]¨]Ð<RÐ,SÐTðð ð
ð 	
r   Fc                 óŒ   — ddl m}  || ||||	¬¦  «        }|j        |j        j        |j        j        |j        |j        dœ}||fS )Nr   )Úmodel_factory)Úsave_quantized)r*   Útext_encoderÚtext_encoder_2ÚvaeÚ	audio_vae)Úkandinsky6_mainre   r*   rg   ÚmodelÚclipri   rj   )Úmodel_filenameÚ
model_typer!   rT   ÚquantizeTransformerÚtext_encoder_quantizationr+   Ú	VAE_dtypeÚmixed_precision_transformerrf   Úsubmodel_no_listÚtext_encoder_filenameÚkwargsre   rl   Úpipes                   r   Ú
load_modelzfamily_handler.load_modelf   su   € ð 	3Ð2Ð2Ð2Ð2Ð2à�˜n¨j¸/ÐK`Ðqð  Añ  Aô  Aˆà Ô,Ø!Ô.Ô4Ø#Ô0Ô5Ø”9Øœð
ð 
ˆð �dˆ{Ðr   c                 ó<   — |                      ddddddœ¦  «         d S )Né
   g      ð?r@   Ú832x480r_   )Únum_inference_stepsÚguidance_scaleÚvideo_lengthÚ
resolutionÚimage_prompt_type)Úupdate)r!   rT   Úui_defaultss      r   Úupdate_default_settingsz&family_handler.update_default_settingsu   s=   € à×ÒØ#%Ø!ØØ#Ø!#ð
ð 
ñ 	ô 	ð 	ð 	ð 	r   r   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r   r   r"   r$   rU   rc   ÚtorchÚbfloat16Úfloat32rx   rƒ   r   r   r   r   r      sW  € € € € € Øð5ð 5ñ „\ð5ð ðð ñ „\ðð ðð ñ „\ðð ð3ð 3ñ „\ð3ð ðð ñ „\ðð ð*ð *ñ „\ð*ð
 ð$
ð $
ñ „\ð$
ðL ð
ð 
ð 
ñ „\ð
ð, Ø_dð  AEð  MRô  M[Ø"œ]ÈÐ^cÐvzð  SWðð ð ñ „\ðð ðð ñ „\ðð ð r   r   )rˆ   Úmodels.help_sharedr   Úshared.utils.hfr   Úpromptsr   r   r   r	   r
   r   r   r   rS   r   r   r   r   ú<module>rŽ      sà   ðà €€€Ø +Ð +Ð +Ð +Ð +Ð +à (Ð (Ð (Ð (Ð (Ð (ð Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uð  Uà#Ð Ø%Ð Ø.Ð ðpð pð pð pð pñ pô pð pð pð pr   