DiffAbXL

AstraZeneca · Oct 12, 2024 · fb06487 · fb06487
1 parent da4b83b
commit fb06487
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/model_utils.py b/utils/model_utils.py
@@ -73,7 +73,7 @@ def __init__(self, config):
         self.residue_emb = ResidueEmbedding(residue_dim, num_atoms, max_aa_types=config['max_aa_types'], max_chain_types=10)
         self.pair_emb = PairEmbedding(pair_dim, num_atoms, max_aa_types=config['max_aa_types'], max_relpos=32)
 
-        # Diffusion model and mask tokens
+        # Diffusion model
         self.diffusion = Diffusion(self.config)
 
     def forward(self, batch):