o *jg@s<ddlmZddlmZddlmZmZddZddZ dS) )gaussian_diffusion)MDM)SpacedDiffusionspace_timestepscCs<|j|dd\}}t|dksJtdd|DsJdS)NF)strictcSsg|]}|dqS)z clip_model.) startswith).0kr m/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/cv/motion_generation/model.py sz&load_model_wo_clip..)Zload_state_dictlenall)modelZ state_dictZ missing_keysZunexpected_keysr r r load_model_wo_clip s  rc Cst dddddddddd|jdd d d d d dd}d}|j}d}d}d}d}td||}tjj} |s8|g}tt||||sDtj j ntj j |sO  tj jntj j| |dddd } || fS)NirTZrot6diiZhml_vecZhumanmlzViT-B/32) ZnjointsZnfeatsZ num_actions translationZpose_repglobZglob_rotZ latent_dimZff_sizesmpl_data_pathZdata_repZdatasetZ clip_versiontextg?Ztensor)Z cond_modeZcond_mask_probZ action_embg?FZcosineg) Z use_timestepsbetasZmodel_mean_typeZmodel_var_type loss_typerescale_timestepsZ lambda_velZ lambda_rcxyzZ lambda_fc)r)rrZ sample_stepsgdZget_named_beta_scheduleZLossTypeZMSErrZ ModelMeanTypeZEPSILONZSTART_XZ ModelVarTypeZ FIXED_LARGEZ FIXED_SMALLZ LEARNED_RANGE) cfgrZpredict_xstartZstepsZ scale_betaZtimestep_respacingZ learn_sigmarrrZ diffusionr r r create_modelsf rN) modulesrrZ modules.mdmrZmodules.respacerrrrr r r r s