sokrypton / ColabDesign

Making Protein Design accessible to all via Google Colab!
529 stars 118 forks source link

Adding solubleMPNN weights #154

Closed KeaunAmani closed 1 year ago

KeaunAmani commented 1 year ago

Added support for solubleMPNN weights to be optionally loaded instead of the default proteinMPNN weights.

Added weights= parameter to the mk_mpnn_model's constructor so that users can specify the weights they want to use.

Weights can be set to either original (default) or soluble to control this behavior.