rsax commited on
Commit
8f87ddc
·
verified ·
1 Parent(s): 1fe1c0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -30,14 +30,14 @@ args.depth = 3
30
  args.block_size = 51
31
 
32
  vqvae_model = vqvae.HumanVQVAE(args,
33
- args.nb_code,
34
- args.code_dim,
35
- args.output_emb_width,
36
- args.down_t,
37
- args.stride_t,
38
- args.width,
39
- args.depth,
40
- args.dilation_growth_rate).to(device)
41
  transformer_model = trans.Text2Motion_Transformer(num_vq=args.nb_code,
42
  embed_dim=1024,
43
  clip_dim=args.clip_dim,
 
30
  args.block_size = 51
31
 
32
  vqvae_model = vqvae.HumanVQVAE(args,
33
+ 512,
34
+ 512,
35
+ 512,
36
+ 2,
37
+ 2,
38
+ 512,
39
+ 3,
40
+ 3).to(device)
41
  transformer_model = trans.Text2Motion_Transformer(num_vq=args.nb_code,
42
  embed_dim=1024,
43
  clip_dim=args.clip_dim,