Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Conversation

marksgraham
Copy link
Collaborator

Fixes #351. Also adds a check that xformers is installed if use_flash_attention=True; previously this check was not done and it errored out if xformers was not installed.

@marksgraham
Copy link
Collaborator Author

Hi @Can-Zhao

I think I've fixed the issue. Could you check this fixes it for you by installing this PR using pip install git+https://github.com/Project-MONAI/GenerativeModels.git@refs/pull/375/merge, and let me know?

@Warvito Warvito merged commit 5e3fe02 into main Apr 20, 2023
@Warvito Warvito deleted the 351_grad_strides_warning_with_ddp branch May 4, 2023 19:06
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

AttentionBlock gives warning of "Grad strides do not match bucket view strides" when using DDP

2 participants