Open xforcevesa opened 5 months ago
Fixed Bug on misalignment between size of pos_embed and input tensor.
pos_embed
Copy the sample_DiT.py to this project and make it successful to run validation, especially with the support of distributed data parallelism.
sample_DiT.py
Description
Fixed Bug on misalignment between size of
pos_embedand input tensor.Copy the
sample_DiT.pyto this project and make it successful to run validation, especially with the support of distributed data parallelism.Changed Files