ó
    È9ig  ã                   ó8   • S SK Jr  S SKJr  S SKJrJr  S rS r	g)é   )Úgaussian_diffusion)ÚMDM)ÚSpacedDiffusionÚspace_timestepsc                 ó¶   • U R                  USS9u  p#[        U5      S:X  d   e[        U Vs/ s H  oDR                  S5      PM     sn5      (       d   eg s  snf )NF)Ústricté    zclip_model.)Úload_state_dictÚlenÚallÚ
startswith)ÚmodelÚ
state_dictÚmissing_keysÚunexpected_keysÚks        Úl/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/models/cv/motion_generation/model.pyÚload_model_wo_clipr   	   s\   € Ø$)×$9Ñ$9Ø˜5ð %:ð %"Ñ!€LäˆÓ 1Ó$Ð$Ð$Ü±\ÓB²\°—‘˜]Ö+±\ÑB×CÑCÐCÑCùÒBs   ­Ac                 ó  • [         SSSSSSSSSSU R                  SS	S
S.SSSS.D6nSnU R                  nSnSnSnSn[        R                  " SX45      n[        R
                  R                  n	U(       d  U/n[        [        X55      UU(       d  [        R                  R                  O[        R                  R                  U(       d  [        R                  R                  O[        R                  R                  U	USSSS9	n
X4$ )NÚ i  r   TÚrot6di   i   Úhml_vecÚhumanmlzViT-B/32)ÚnjointsÚnfeatsÚnum_actionsÚtranslationÚpose_repÚglobÚglob_rotÚ
latent_dimÚff_sizeÚsmpl_data_pathÚdata_repÚdatasetÚclip_versionÚtextgš™™™™™¹?Útensor)Ú	cond_modeÚcond_mask_probÚ
action_embg      ð?FÚcosineg        )	Úuse_timestepsÚbetasÚmodel_mean_typeÚmodel_var_typeÚ	loss_typeÚrescale_timestepsÚ
lambda_velÚlambda_rcxyzÚ	lambda_fc)r   )r   r#   Úsample_stepsÚgdÚget_named_beta_scheduleÚLossTypeÚMSEr   r   ÚModelMeanTypeÚEPSILONÚSTART_XÚModelVarTypeÚFIXED_LARGEÚFIXED_SMALLÚLEARNED_RANGE)Úcfgr   Úpredict_xstartÚstepsÚ
scale_betaÚtimestep_respacingÚlearn_sigmar2   r.   r1   Ú	diffusions              r   Úcreate_modelrI      s  € ÜØ
ðàØØØØØØØØØ×)Ñ)ØØØñð   Ø!Ø"ñ
ñ€Eð* €NØ×Ñ€EØ€JØÐØ€KØÐä×&Ò& x°ÓC€EÜ—‘—‘€IæØ#˜WÐäÜ% eÓ@Øæ .ô ×)Ñ)×1Ò1Ü46×4DÑ4D×4LÑ4Lö  +ô +-¯/©/×*EÒ*EÜ02·±×0MÑ0MØØ+ØØØñ€Ið ÐÐó    N)
Úmodulesr   r7   Úmodules.mdmr   Úmodules.respacer   r   r   rI   © rJ   r   Ú<module>rO      s   ðõ .Ý ß =òDó1rJ   