ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removed wan submodule stuff for now

Browse Source
This commit is contained in:
Jaret Burkett
2025-03-04 00:32:24 -07:00
parent 8bb47d1bfe
commit c57434ad7b
1 changed files with 1 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
toolkit/models/wan21.py
View File

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