Update config.py

Author: sokrypton

colabdesign/af/alphafold/model/config.py

@@ -143,8 +143,8 @@ def model_config(name: str) -> ml_collections.ConfigDict:
                 'prev_pos': [NUM_RES, None, None],
                 'prev_pair': [NUM_RES, NUM_RES, None],
                 'prev_msa_first_row': [NUM_RES, None],
-                'rm_template_seq': [NUM_RES, None],
-                'rm_template_sc': [NUM_RES, None]
+                'rm_template_seq': [NUM_RES],
+                'rm_template_sc': [NUM_RES]
             },
         },
     },