Spaces:
Runtime error
Runtime error
Update wan/text2video.py
Browse files- 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.
|
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,
|