Update app.py
Browse files
app.py
CHANGED
@@ -30,14 +30,14 @@ args.depth = 3
|
|
30 |
args.block_size = 51
|
31 |
|
32 |
vqvae_model = vqvae.HumanVQVAE(args,
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
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,
|