Spaces:
Running
on
Zero
Running
on
Zero
Update wan/modules/model.py
Browse files- wan/modules/model.py +1 -1
wan/modules/model.py
CHANGED
@@ -8,7 +8,7 @@ from diffusers.configuration_utils import ConfigMixin, register_to_config
|
|
8 |
from diffusers.models.modeling_utils import ModelMixin
|
9 |
|
10 |
from .attention import flash_attention
|
11 |
-
from
|
12 |
__all__ = ['WanModel']
|
13 |
|
14 |
|
|
|
8 |
from diffusers.models.modeling_utils import ModelMixin
|
9 |
|
10 |
from .attention import flash_attention
|
11 |
+
from .communications import gather_sequence,split_sequence,all_to_all_4D
|
12 |
__all__ = ['WanModel']
|
13 |
|
14 |
|