py-img-gen
/

ncsn-mnist

Model card Files Files and versions Community

shunk031 commited on 9 days ago

Commit

64789a5

•

1 Parent(s): bf85952

Update model_index.json

Files changed (1) hide show

model_index.json +3 -3

model_index.json CHANGED Viewed

@@ -1,12 +1,12 @@
 {
-  "_class_name": "NCSNPipeline",
   "_diffusers_version": "0.32.0",
   "scheduler": [
-    "ncsn.scheduler.scheduling_ncsn",
     "AnnealedLangevinDynamicsScheduler"
   ],
   "unet": [
-    "ncsn.unet.unet_2d_ncsn",
     "UNet2DModelForNCSN"
   ]
 }

 {
+  "_class_name": ["pipeline_ncsn", "NCSNPipeline"],
   "_diffusers_version": "0.32.0",
   "scheduler": [
+    "scheduling_ncsn",
     "AnnealedLangevinDynamicsScheduler"
   ],
   "unet": [
+    "unet_2d_ncsn",
     "UNet2DModelForNCSN"
   ]
 }