openfold icon indicating copy to clipboard operation
openfold copied to clipboard

Can not load AlphaFold multimer checkpoint params_model_[1-5]_multimer.npz for inference

Open bing-song opened this issue 2 years ago • 5 comments

Can not load multimer checkpoint in the main branch for both Docker container and ENV.

Here is command

Screen Shot 2022-08-08 at 11 08 06 AM

Here is the error message

Screen Shot 2022-08-08 at 11 08 26 AM

bing-song avatar Aug 08 '22 18:08 bing-song

This issue may be the same as issue #110

bing-song avatar Aug 08 '22 18:08 bing-song

More information

The inference works fine when I use AlphaFold models params_model_[1-5].npz (not multimer model).

bing-song avatar Aug 08 '22 19:08 bing-song

In main branch openfold:config.py, it does not have

elif "multimer" in name:
        c.globals.is_multimer = True
        for k,v in multimer_model_config_update.items():
            c.model[k] = v

However, this line is in multimer branch.

My question is that should we use multimer branch or main branch to load a multimer model?

bing-song avatar Aug 08 '22 20:08 bing-song

For now, you should definitely use the hacky AlphaFold-Gap implementation on the main branch. The multimer branch is still experimental (I've had zero time to work on that recently), and while I'd say it's nearly done, there are still some huge blemishes like this.

gahdritz avatar Aug 08 '22 21:08 gahdritz

@gahdritz can I ask, have you looked at what the gap trick does to the template and MSA feature retrieval / computation? Are they primarily from the longer sequence or anything like that? If I understand correctly, the pipeline with or without linker is the same in terms of MSA computation and template retrieval?

gieses avatar Feb 06 '23 12:02 gieses