Skip to content

Commit

Permalink
todo
Browse files Browse the repository at this point in the history
  • Loading branch information
@lucidrains
lucidrains authored Jun 18, 2023
1 parent fa58f5c commit 875bd25
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -137,6 +137,7 @@ video_as_images_out = unet(video, enable_time = False)
- [x] give attention the best positional embeddings research has to offer
- [x] soup up the attention

- [ ] add flash attention
- [ ] make sure dalle2-pytorch can accept `SpaceTimeUnet` for training

## Citations
Expand Down

0 comments on commit 875bd25

Please sign in to comment.