Spaces:
Running
Running
File size: 1,268 Bytes
a0d91d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
from mogen.models import build_submodule
def get_motion_model(name, ckpt_path):
if name == 'kit_ml':
model = build_submodule(dict(
type='T2MMotionEncoder',
input_size=251,
movement_hidden_size=512,
movement_latent_size=512,
motion_hidden_size=1024,
motion_latent_size=512,
))
else:
model = build_submodule(dict(
type='T2MMotionEncoder',
input_size=263,
movement_hidden_size=512,
movement_latent_size=512,
motion_hidden_size=1024,
motion_latent_size=512,
))
model.load_pretrained(ckpt_path)
return model
def get_text_model(name, ckpt_path):
if name == 'kit_ml':
model = build_submodule(dict(
type='T2MTextEncoder',
word_size=300,
pos_size=15,
hidden_size=512,
output_size=512,
max_text_len=20
))
else:
model = build_submodule(dict(
type='T2MTextEncoder',
word_size=300,
pos_size=15,
hidden_size=512,
output_size=512,
max_text_len=20
))
model.load_pretrained(ckpt_path)
return model
|