shi-labs
/

versatile-diffusion

fix: ValueError: text_unet/versatile_diffusion.py as defined in `model_index.json` does not exist in shi-labs/versatile-diffusion and is not a module in 'diffusers/pipelines'

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",