From dcc090138281c26b70409367d4e3ed08558d7690 Mon Sep 17 00:00:00 2001
From: YuxuanCai <LarryTsai@users.noreply.huggingface.co>
Date: Mon, 21 Oct 2024 00:47:09 +0000
Subject: [PATCH] Update model_index.json

---
 model_index.json | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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"