Add conditions to model constructor

openfold/model/model.py

@@ -79,6 +79,8 @@ class AlphaFold(nn.Module):
         self.recycling_embedder = RecyclingEmbedder(
             **self.config["recycling_embedder"],
         )
+        
+        if(self.template_config.enabled):
             self.template_angle_embedder = TemplateAngleEmbedder(
                 **self.template_config["template_angle_embedder"],
             )
@@ -91,12 +93,15 @@ class AlphaFold(nn.Module):
             self.template_pointwise_att = TemplatePointwiseAttention(
                 **self.template_config["template_pointwise_attention"],
             )
+       
+        if(self.extra_msa_config.enabled):
             self.extra_msa_embedder = ExtraMSAEmbedder(
                 **self.extra_msa_config["extra_msa_embedder"],
             )
             self.extra_msa_stack = ExtraMSAStack(
                 **self.extra_msa_config["extra_msa_stack"],
             )
+        
         self.evoformer = EvoformerStack(
             **self.config["evoformer_stack"],
         )