Skip to content

Remove xformers install#781

Merged
pstjohn merged 3 commits into
NVIDIA-BioNeMo:mainfrom
pstjohn:pstjohn/remove-xformers
Mar 24, 2025
Merged

Remove xformers install#781
pstjohn merged 3 commits into
NVIDIA-BioNeMo:mainfrom
pstjohn:pstjohn/remove-xformers

Conversation

@pstjohn
Copy link
Copy Markdown
Collaborator

@pstjohn pstjohn commented Mar 24, 2025

Removes the xformers install in the framework container, which is only used for running the huggingface AMPLIFY baselines.

Comment thread sub-packages/bionemo-amplify/tests/bionemo/amplify/test_hf_rotary.py Outdated
pstjohn added 3 commits March 24, 2025 10:33
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
@pstjohn pstjohn force-pushed the pstjohn/remove-xformers branch from 3d11a2d to 68f94db Compare March 24, 2025 17:33
@pstjohn pstjohn enabled auto-merge March 24, 2025 18:15
@pstjohn pstjohn added this pull request to the merge queue Mar 24, 2025
Merged via the queue into NVIDIA-BioNeMo:main with commit d8e1a62 Mar 24, 2025
@pstjohn pstjohn deleted the pstjohn/remove-xformers branch March 24, 2025 20:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants