diffusers
e1d508ae - [Tests] Improve transformers model test suite coverage - Latte (#8919)

Commit
1 year ago
[Tests] Improve transformers model test suite coverage - Latte (#8919) * add LatteTransformer3DModel model test * change patch_size to 1 * reduce req len * reduce channel dims * increase num_layers * reduce dims further * run make style --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: Aryan <aryan@huggingface.co>
Author
Parents
Loading