Error when using --save-single or pair-representations argument : NoneType object is not subscriptable

[bayarovici]

Hi Everyone. Whenever I try to save embeddings for my protein sequence I get an error suggesting nothing is returned from running the model.

Running the multimer model on a protein complex (in csv file format) does not work: colabfold_batch --amber --templates --numer_recycle 3 model_type=AlphaFold2-multimer-v2 --save-pair-representations protein_complex.csv output_folder

However, running the simple model on a monomer fasta file works as expected: colabfold_batch --amber --templates --number_recycle 3 --save-pair-representations monomer.fasta output_folder

Does save-pair-representations or single-representations not work for protein complexes or is there a bug?

[lucidrains]

@bayarovici :wave: glad to see you using this feature! i think https://github.com/sokrypton/ColabFold/pull/180 should address alphafold multimers, but you may want to checkout the branch and do some testing first