From c57434ad7b6c2f672e91c42c4679966d47c10c84 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Tue, 4 Mar 2025 00:32:24 -0700
Subject: [PATCH] Removed wan submodule stuff for now

---
 toolkit/models/wan21.py | 15 +--------------
 1 file changed, 1 insertion(+), 14 deletions(-)

diff --git a/toolkit/models/wan21.py b/toolkit/models/wan21.py
index f30c1cd6..b9a98400 100644
--- a/toolkit/models/wan21.py
+++ b/toolkit/models/wan21.py
@@ -1,4 +1,4 @@
-
+# WIP, coming soon ish
 import torch
 from toolkit.config_modules import GenerateImageConfig, ModelConfig
 from toolkit.models.base_model import BaseModel
@@ -23,19 +23,6 @@ import torch.distributed as dist
 from tqdm import tqdm
 
 
-WAN_ROOT = os.path.join(REPOS_ROOT, "wan21")
-sys.path.append(WAN_ROOT)
-
-if True:
-    from wan.text2video import WanT2V
-    from wan.distributed.fsdp import shard_model
-    from wan.modules.model import WanModel
-    from wan.modules.t5 import T5EncoderModel
-    from wan.modules.vae import WanVAE
-    from wan.utils.fm_solvers import (FlowDPMSolverMultistepScheduler,
-                                get_sampling_sigmas, retrieve_timesteps)
-    from wan.utils.fm_solvers_unipc import FlowUniPCMultistepScheduler
-
 class Wan21(BaseModel):
     def __init__(
             self,