Skip to content

Support DiT Trainer #533

@Coach257

Description

@Coach257

Feature request

  1. Refactor branch dit to use trainer. Training wan2.1 as example.
  2. add vae & text encoder of wan2.1 and support wan2.1 inference.

Motivation

Currently, training wan2.1 is not friendly in VeOmni.

Your contribution

Support training and inference wan2.1.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions