Update model_index.json

model_index.json

@@ -7,6 +7,7 @@
   ],
   "unet": [
     "diffusers",
-    "UNet2DModel"
+    "UNet2DModel",
+    "diffusion_pytorch_model.safetensors"
   ]
 }