diffusers
[UNet2DConditionModel] add an option to upcast attention to fp32
#1590
Merged

[UNet2DConditionModel] add an option to upcast attention to fp32 #1590

patrickvonplaten merged 1 commit into main from upcast-attention
patil-suraj
patil-suraj upcast attention
db191742
patil-suraj patil-suraj requested a review from patrickvonplaten patrickvonplaten 3 years ago
HuggingFaceDocBuilderDev
pcuenca
pcuenca approved these changes on 2022-12-07
patrickvonplaten
patrickvonplaten approved these changes on 2022-12-07
patrickvonplaten patrickvonplaten merged 170ebd28 into main 3 years ago
patrickvonplaten patrickvonplaten deleted the upcast-attention branch 3 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone