mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-06-03 22:04:53 +00:00
18d53feb7a
* Add VACE * Support training with multiple gpus * Update default args for vace task * vace block update * Add vace exmaple jpg * Fix dist vace fwd hook error * Update vace exmample * Update vace args * Update pipeline name for vace * vace gradio and Readme * Update vace snake png --------- Co-authored-by: hanzhn <han.feng.jason@gmail.com>
19 lines
422 B
Python
19 lines
422 B
Python
from .attention import flash_attention
|
|
from .model import WanModel
|
|
from .t5 import T5Decoder, T5Encoder, T5EncoderModel, T5Model
|
|
from .tokenizers import HuggingfaceTokenizer
|
|
from .vace_model import VaceWanModel
|
|
from .vae import WanVAE
|
|
|
|
__all__ = [
|
|
'WanVAE',
|
|
'WanModel',
|
|
'VaceWanModel',
|
|
'T5Model',
|
|
'T5Encoder',
|
|
'T5Decoder',
|
|
'T5EncoderModel',
|
|
'HuggingfaceTokenizer',
|
|
'flash_attention',
|
|
]
|