diff --git a/model_index.json b/model_index.json
index f978153..48de5d9 100644
--- a/model_index.json
+++ b/model_index.json
@@ -1,7 +1,9 @@
 {
-  "_class_name": "AllegroPipeline",
+  "_class_name":  [
+    "pipeline_allegro",
+    "AllegroPipeline"
+  ],
   "_diffusers_version": "0.30.3",
-  "_name_or_path": "/cpfs/data/user/larrytsai/Projects/Yi-VG/allegro",
   "scheduler": [
     "diffusers",
     "EulerAncestralDiscreteScheduler"