nyarunyarunya commited on
Commit
9384a5d
·
verified ·
1 Parent(s): 2fd9c9c

Update wan/modules/__init__.py

Browse files
Files changed (1) hide show
  1. wan/modules/__init__.py +1 -1
wan/modules/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
  from .attention import flash_attention
2
- from .model import WanModel
3
  from .t5 import T5Decoder, T5Encoder, T5EncoderModel, T5Model
4
  from .tokenizers import HuggingfaceTokenizer
5
  from .vae import WanVAE
 
1
  from .attention import flash_attention
2
+ from .model_infer import WanModel
3
  from .t5 import T5Decoder, T5Encoder, T5EncoderModel, T5Model
4
  from .tokenizers import HuggingfaceTokenizer
5
  from .vae import WanVAE