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