diffusers
f848feba
- feat: allow sharding for auraflow. (#8853)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
324 days ago
feat: allow sharding for auraflow. (#8853)
References
#8853 - feat: allow sharding for auraflow.
Author
sayakpaul
Parents
b3825500
Files
2
src/diffusers/models/transformers
auraflow_transformer_2d.py
tests/models/transformers
test_models_transformer_aura_flow.py
Loading
test_models_transformer_aura_flow.py