ZeyuLing commited on
Commit
16f719f
·
verified ·
1 Parent(s): a0329ac

Normalize Motius Pipeline.from_pretrained metadata

Browse files
Files changed (2) hide show
  1. README.md +8 -0
  2. model_index.json +22 -1
README.md CHANGED
@@ -87,3 +87,11 @@ motion tokens, and the reciprocal Transformer translates those tokens to text.
87
  year={2022}
88
  }
89
  ```
 
 
 
 
 
 
 
 
 
87
  year={2022}
88
  }
89
  ```
90
+
91
+ ## Direct Loading
92
+
93
+ ```python
94
+ from motius import Pipeline
95
+
96
+ pipeline = Pipeline.from_pretrained("ZeyuLing/Motius-TM2T-HumanML3D")
97
+ ```
model_index.json CHANGED
@@ -5,5 +5,26 @@
5
  "format": "motius-tm2t-artifact-v1",
6
  "bundle_class": "motius.models.tm2t.TM2TBundle",
7
  "pipeline_class": "motius.pipelines.tm2t.TM2TPipeline",
8
- "api": "motius.pipelines.tm2t.TM2TPipeline.from_pretrained"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  }
 
5
  "format": "motius-tm2t-artifact-v1",
6
  "bundle_class": "motius.models.tm2t.TM2TBundle",
7
  "pipeline_class": "motius.pipelines.tm2t.TM2TPipeline",
8
+ "api": {
9
+ "loader": "motius.Pipeline.from_pretrained",
10
+ "task_methods": [
11
+ "infer_motion_to_text"
12
+ ]
13
+ },
14
+ "format_version": 1,
15
+ "tasks": [
16
+ "motion_to_text"
17
+ ],
18
+ "required_files": [
19
+ "glove/our_vab_data.npy",
20
+ "glove/our_vab_idx.pkl",
21
+ "glove/our_vab_words.pkl",
22
+ "t2m/M2T_EL4_DL4_NH8_PS/meta/mean.npy",
23
+ "t2m/M2T_EL4_DL4_NH8_PS/meta/std.npy",
24
+ "t2m/M2T_EL4_DL4_NH8_PS/model/finest.tar",
25
+ "t2m/M2T_EL4_DL4_NH8_PS/opt.txt",
26
+ "t2m/VQVAEV3_CB1024_CMT_H1024_NRES3/meta/mean.npy",
27
+ "t2m/VQVAEV3_CB1024_CMT_H1024_NRES3/meta/std.npy",
28
+ "t2m/VQVAEV3_CB1024_CMT_H1024_NRES3/model/finest.tar"
29
+ ]
30
  }