diff --git a/model_index.json b/model_index.json
index 48de5d9..5e152b7 100644
--- a/model_index.json
+++ b/model_index.json
@@ -17,11 +17,11 @@
     "T5Tokenizer"
   ],
   "transformer": [
-    "diffusers_modules.local.transformer_3d_allegro",
+    "transformer_3d_allegro",
     "AllegroTransformer3DModel"
   ],
   "vae": [
-    "diffusers_modules.local.vae_allegro",
+    "vae_allegro",
     "AllegroAutoencoderKL3D"
   ]
 }