Added Support for WAN Image Generation

api/transformerlab/plugins/image_diffusion/index.json

@@ -7,15 +7,9 @@
   "version": "0.0.5",
   "git": "",
   "url": "",
-  "files": [
-    "main.py",
-    "diffusion_worker.py",
-    "setup.sh"
-  ],
-  "supported_hardware_architectures": [
-    "cuda",
-    "amd"
-  ],
+  "files": ["main.py", "diffusion_worker.py", "setup.sh"],
+  "model_architectures": ["DiffusionPipeline", "WanPipeline"],
+  "supported_hardware_architectures": ["cuda", "amd"],
   "_dataset": false,
   "setup-script": "setup.sh",
   "parameters": {},

api/transformerlab/plugins/image_diffusion/setup.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+uv pip install git+https://github.com/huggingface/diffusers transformers accelerate safetensors
+uv pip install peft

api/transformerlab/plugins/video_diffusion_plugin/index.json

@@ -0,0 +1,33 @@
+{
+  "name": "Video Diffusion",
+  "uniqueId": "video_diffusion",
+  "description": "Generate videos in the Diffusion tab using this plugin",
+  "plugin-format": "python",
+  "type": "diffusion",
+  "version": "0.0.6",
+  "git": "",
+  "url": "",
+  "files": ["main.py", "video_diffusion_worker.py", "setup.sh"],
+  "model_architectures": [
+    "DiffusionPipeline",
+    "WanPipeline",
+    "StableVideoDiffusionPipeline",
+    "WanImageToVideoPipeline"
+  ],
+  "supported_hardware_architectures": ["cuda", "amd"],
+  "_dataset": false,
+  "setup-script": "setup.sh",
+  "parameters": {
+    "tflabcustomui_is_text2video": {
+      "type": "boolean",
+      "title": "Text-to-Video Generation",
+      "description": "If checked, generates a video from a text prompt. Otherwise, an input image is required for image-to-video.",
+      "default": false
+    }
+  },
+  "parameters_ui": {
+    "tflabcustomui_is_text2video": {
+      "ui:widget": "checkbox"
+    }
+  }
+}