nyarunyarunya commited on
Commit
419e457
·
verified ·
1 Parent(s): 1e48c5d

Update wan/text2video.py

Browse files
Files changed (1) hide show
  1. wan/text2video.py +1 -1
wan/text2video.py CHANGED
@@ -15,7 +15,7 @@ import torch.distributed as dist
15
  from tqdm import tqdm
16
 
17
  from .distributed.fsdp import shard_model
18
- from .modules.model import WanModel
19
  from .modules.t5 import T5EncoderModel
20
  from .modules.vae import WanVAE
21
  from .utils.fm_solvers import (FlowDPMSolverMultistepScheduler,
 
15
  from tqdm import tqdm
16
 
17
  from .distributed.fsdp import shard_model
18
+ from .modules.model_infer import WanModel
19
  from .modules.t5 import T5EncoderModel
20
  from .modules.vae import WanVAE
21
  from .utils.fm_solvers import (FlowDPMSolverMultistepScheduler,