Add ESMFold #19977

sgugger merged 46 commits into main from add_esmfold
Rocketknight1
HuggingFaceDocBuilderDev
Rocketknight1 Rocketknight1 marked this pull request as ready for review 3 years ago
Rocketknight1 initial commit
5fdd2724
Rocketknight1 First draft that gets outputs without crashing!
2996fce2
Rocketknight1 Add all the ported openfold dependencies
74229d26
Rocketknight1 testing
45773746
Rocketknight1 Restructure config files for ESMFold
f1641cc5
Rocketknight1 Debugging to find output discrepancies
65569f03
sgugger Mainly style
26f3e25d
sgugger Make model runnable without extra deps
988752ac
sgugger Remove utils and merge them to the modeling file
92a58871
Rocketknight1 Use correct gelu and remove some debug prints
dd1f4808
sgugger More cleanup
6e31a39a
Rocketknight1 Update esm docs
0e69779d
Rocketknight1 Update conversion script to support ESMFold properly
010fab68
Rocketknight1 Port some top-level changes from ESMFold repo
94d06633
Rocketknight1 Expand EsmFold docstrings
728f21f6
Rocketknight1 Make attention_mask optional (default to all 1s)
28839b7f
sgugger sgugger force pushed to 28839b7f 3 years ago
Rocketknight1 Add inference test for ESMFold
48c97f44
sgugger Use config and not n kwargs
2422d11c
sgugger Merge branch 'add_esmfold' of github.com:huggingface/transformers int…
fcbf85d5
sgugger Add modeling output class
e7bf6a5e
sgugger Remove einops
69d51694
Rocketknight1 Remove chunking in ESM FFN
5770297b
Rocketknight1 Update tests for ESMFold
f8a9945a
sgugger Quality
6ab675ca
sgugger Merge branch 'add_esmfold' of github.com:huggingface/transformers int…
88757cba
sgugger REpo consistency
1ead4c0f
Rocketknight1 Remove tree dependency from ESMFold
cff0224a
Rocketknight1 Merge remote-tracking branch 'origin/add_esmfold' into add_esmfold
b83a592e
Rocketknight1 make fixup
5b0fbaef
Rocketknight1 Add an error in case my structure map function breaks later
bac51f28
sgugger Remove needless code
61d65810
sgugger Fix merge conflicts
44ed50ff
Rocketknight1 Stop auto-casting the LM to float16 so CPU tests pass
f5e75754
Rocketknight1 Stop auto-casting the LM to float16 so CPU tests pass
8bbf3759
Rocketknight1 Final test updates
7632a128
sgugger Split test file
d14fddbc
sgugger Copyright and quality
a91465b4
sgugger Unpin PyTorch to see built doc
e45e6fc6
Rocketknight1 Fix config file to_dict() method
7bebbbff
Rocketknight1 Add some docstrings to the output
60d681c3
Rocketknight1 Skip TF checkpoint tests for ESM until we reupload those
e2d9ff7f
Rocketknight1 make fixup
e765f599
Rocketknight1 More docstrings
9c9f9fa8
sgugger Unpin to get even with main
fd455c4a
sgugger Merge branch 'add_esmfold' of github.com:huggingface/transformers int…
020a3022
sgugger Flag example to write
f6f4a2c4
sgugger
sgugger sgugger merged 7f9b7b3f into main 3 years ago
sgugger sgugger deleted the add_esmfold branch 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone