shi-labs
/

versatile-diffusion

VersatileDiffusionPipeline

image-variation

Model card Files Files and versions

Update model_index.json

#7

by nirajan111 - opened Jan 24, 2025

base: refs/heads/main

←

from: refs/pr/7

Discussion Files changed

Files changed (1) hide show

model_index.json +2 -2

model_index.json CHANGED Viewed

@@ -22,8 +22,8 @@
     "CLIPTextModelWithProjection"
   ],
   "text_unet": [
-    "versatile_diffusion",
-    "UNetFlatConditionModel"
   ],
   "tokenizer": [
     "transformers",

     "CLIPTextModelWithProjection"
   ],
   "text_unet": [
+    "diffusers",
+    "UNet2DConditionModel"
   ],
   "tokenizer": [
     "transformers",