finetuned models support

2025-06-12 10:00:47 +02:00
parent 43aa414eaf
commit 3749d23d44
23 changed files with 779 additions and 121 deletions
--- a/configs/vace_14B.json
+++ b/configs/vace_14B.json
@@ -0,0 +1,16 @@
+{
+  "_class_name": "VaceWanModel",
+  "_diffusers_version": "0.30.0",
+  "dim": 5120,
+  "eps": 1e-06,
+  "ffn_dim": 13824,
+  "freq_dim": 256,
+  "in_dim": 16,
+  "model_type": "t2v",
+  "num_heads": 40,
+  "num_layers": 40,
+  "out_dim": 16,
+  "text_len": 512,
+  "vace_layers": [0, 5, 10, 15, 20, 25, 30, 35],
+  "vace_in_dim": 96
+}