Add Uni-Fold Conformer (a diffusion-based conformation sampling method)#148

Open

PKUfjh wants to merge 26 commits intodptech-corp:mainfrom

PKUfjh commented May 18, 2024

Add ufconf model code.
Add ufconf inference scripts.
Add ufconf README file

PKUfjh added 3 commits

May 18, 2024 13:45


          organize scripts for ufconf

8825e7d


          update unifold module

9c73cea


          set default ufconf model, update readme

7967d31

ZiyaoLi reviewed

View reviewed changes

unifold/msa/mmcif.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

README.md Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

example_diffold/1ake_dataset_monomer.json Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

finetune_ufconf.sh Outdated Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

notebooks/README.md Outdated Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

notebooks/run_ufconf_denoise.py

Member

ZiyaoLi May 21, 2024

these 3 py files are noqa by me currently

Author

PKUfjh May 23, 2024

refined the scripts

ZiyaoLi reviewed

View reviewed changes

setup.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

ufconf/README.md Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

ufconf/config.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

ufconf/dataset.py Outdated Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

ufconf/dataset_clu.py Outdated Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

ufconf/diffold.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

ufconf/diffusion/angle.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

ufconf/task.py Outdated Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

ufconf/utils.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

unifold/config.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

unifold/data/lmdb_dataset.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

unifold/dataset.py

Member

ZiyaoLi May 21, 2024

revert this

Author

PKUfjh May 23, 2024

this has problem running the inference scripts, need to discuss how to refine this part of the codes

ZiyaoLi reviewed

View reviewed changes

unifold/inference.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

unifold/inference_symmetry.py Show resolved Hide resolved

ZiyaoLi reviewed

View reviewed changes

unifold/data/msa_pairing.py

                   "rigid_group_default_frame",
                   # zy
                   "num_sym",
+                  "input_atom_positions",

Member

ZiyaoLi May 21, 2024

these 4 keywords seem never used. try removing these and see if still runnable

ZiyaoLi reviewed

View reviewed changes

unifold/refine/modules/fold.py Outdated Show resolved Hide resolved

ZiyaoLi requested changes

View reviewed changes

unifold/symmetry/dataset.py Show resolved Hide resolved

unifold/task.py Show resolved Hide resolved

ZiyaoLi changed the title ~~Add ufconf scripts~~ Add Uni-Fold Conformer (a diffusion-based conformation sampling method)

ZiyaoLi changed the base branch from main to ufconformer

May 21, 2024 10:28

ZiyaoLi changed the base branch from ufconformer to main

May 21, 2024 10:30


          remove redundant codes

cc2df8b

PKUfjh added 4 commits

May 22, 2024 19:45


          refine inference scripts, update readme

793b183


          revert refine and symmetry part of unifold

b6c66b7


          fix monomer bugs

d69a9e7


          rename diffold to ufconf

e1d95e0

PKUfjh requested a review from ZiyaoLi

May 23, 2024 10:17


          add test script

4b91957

PKUfjh force-pushed the ufconf branch from 49fe427 to 4b91957 Compare

May 29, 2024 14:55

PKUfjh and others added 17 commits

May 29, 2024 22:56


          add list selection for motif

82c7fb7


          add RAC-47 dataset

56ec119


          add fasta file input option

5113f24


          minor fix

68cc4b0


          update config

daff8d1


          update inference script

bde0899


          update readme

d7d06d2


          fix multimer bugs for fasta input

7eae6f6


          fix partial generation bugs

5bb8619


          update msa generation code

bcc7a86


          update msa example

b25c02a


          fix msa generation bugs

1ed80d2


          fix label bug


          add checkpoint info

eb5f20e


          update installation guide

85a6727


          add multimer example

a7e9e6b


          update ufconf readme

c852827

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet