Add diffusionkit into setup.py

junpeiz · web-flow · commit 5d5f15551229 · 2024-08-26T15:29:13.000-07:00
Adds diffusionkit to package installation dependency to avoid erroring out when converting UNet.
diff --git a/setup.py b/setup.py
@@ -26,6 +26,7 @@
         "invisible-watermark",
         "safetensors",
         "matplotlib",
+        "diffusionkit",
     ],
     packages=find_packages(),
     classifiers=[
