This repository was archived by the owner on Feb 7, 2025. It is now read-only.
This repository was archived by the owner on Feb 7, 2025. It is now read-only.
Add option to use memory efficient attention to the Diffusion UNet #210
Description
As suggested by @ericspod and @danieltudosiu , the DiffusionModelUNet should have an argument defining if it will use the memory efficient attention from cformers or not.
Metadata
Metadata
Assignees
Labels
No labels