Bundle CLIP for offline Pipeline.from_pretrained

clip.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:431adb635d7a35aaaf5b70acf8738712c6f9ae30d08e17698c6f505c9a686fb8
+size 353470708

mdm_config.json

@@ -1,6 +1,14 @@
 {
   "model_type": "mdm",
+  "format": "motius-mdm-v2",
   "guidance_param": 2.5,
+  "components": {
+    "clip": {
+      "name": "ViT-B/32",
+      "stored_in_artifact": true,
+      "path": "clip.safetensors"
+    }
+  },
   "config": {
     "dataset": "humanml",
     "arch": "trans_enc",

model_index.json

@@ -1,23 +1,38 @@
 {
   "_class_name": "MDMPipeline",
   "_library_name": "motius",
-  "format_version": 1,
-  "pipeline_class": "motius.pipelines.mdm.MDMPipeline",
-  "bundle_class": "motius.models.mdm.MDMBundle",
+  "model_type": "mdm",
+  "format": "motius-mdm-v2",
+  "pipeline_class": "motius.pipelines.mdm.pipeline.MDMPipeline",
+  "bundle_class": "motius.models.mdm.bundle.MDMBundle",
   "tasks": [
     "text_to_motion"
   ],
   "required_files": [
-    "Mean.npy",
-    "Std.npy",
     "mdm_config.json",
-    "model.safetensors"
+    "model.safetensors",
+    "clip.safetensors",
+    "Mean.npy",
+    "Std.npy"
   ],
+  "artifacts": {
+    "model": "model.safetensors",
+    "clip": "clip.safetensors",
+    "mean": "Mean.npy",
+    "std": "Std.npy"
+  },
+  "components": {
+    "clip": {
+      "name": "ViT-B/32",
+      "stored_in_artifact": true,
+      "path": "clip.safetensors"
+    }
+  },
   "api": {
     "loader": "motius.Pipeline.from_pretrained",
     "task_methods": [
-      "infer_text_to_motion"
+      "infer_text_to_motion",
+      "infer_t2m"
     ]
-  },
-  "artifact_format": "motius-mdm-v1"
-}
+  }
+}