Commit Graph

6 Commits

Author SHA1 Message Date
Angelos Katharopoulos
c109d9b596 Fix the seed for data parallel 2025-03-22 16:50:28 -07:00
Angelos Katharopoulos
a1e259607e Fix data parallel generation 2025-03-22 16:43:40 -07:00
Angelos Katharopoulos
208856520d Update for nn.layers.distributed 2025-03-22 16:35:32 -07:00
Angelos Katharopoulos
7fbd1619eb Initial working distributed flux 2025-03-22 15:02:39 -07:00
Zak B. Elep
3d62b058a4
fix: typo on flux model preloading (#1050) 2024-10-15 09:13:01 -07:00
Angelos Katharopoulos
a5f2bab070
Add FLUX finetuning (#1028) 2024-10-11 21:17:41 -07:00