PR #10515 implementing flux on TPUs with ptxla

implementing flux on TPUs with ptxla #10515

yiyixuxu merged 13 commits into huggingface:main from entrpn:flux_ptxla_trillium

implementing flux on TPUs with ptxla

2816222f

Merge branch 'main' into flux_ptxla_trillium

8c425427

hlky commented on 2025-01-10

Merge branch 'main' into flux_ptxla_trillium

fdc592e3

Merge branch 'main' into flux_ptxla_trillium

83b55ba1

add xla flux attention class

378f7ed4

run make style/quality

ad1e1cf3

Merge branch 'main' into flux_ptxla_trillium

e3ce2ba7

yiyixuxu commented on 2025-01-13

yiyixuxu added close-to-merge

Update src/diffusers/models/attention_processor.py

2d7c1985

yiyixuxu commented on 2025-01-13

Update src/diffusers/models/attention_processor.py

e35dd67f

Merge branch 'main' into flux_ptxla_trillium

d13df4fb

Merge branch 'main' into flux_ptxla_trillium

4ae06877

run style and quality

40b154ab

Merge branch 'main' into flux_ptxla_trillium

820027f4

yiyixuxu approved these changes on 2025-01-16

yiyixuxu merged aeac0a00 into main 260 days ago

Reviewers

yiyixuxu

hlky

Assignees

No one assigned

Labels

close-to-merge

Milestone

No milestone