Federated Learning of diffusion models using the Flower framework. Currently supports training of DDPM on CINIC-10 dataset using FedAvg strategy.
TODO:
- Add FedProx strategy and compare results
- Add DiT model training
colinlaganier/FederatedDiffusionModels
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|