Update README.md

kyegomez · May 9, 2024 · 1ba2753 · 1ba2753
1 parent 455ffbd
commit 1ba2753
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -35,10 +35,10 @@ Need review but basically it operates on atomic coordinates.
 
 ```python
 import torch
-from alphafold3.diffusion import GeneticDiffusionModuleBlock
+from alphafold3.diffusion import GeneticDiffusion
 
 # Create an instance of the GeneticDiffusionModuleBlock
-model = GeneticDiffusionModuleBlock(channels=3, training=True)
+model = GeneticDiffusion(channels=3, training=True)
 
 # Generate random input coordinates
 input_coords = torch.randn(10, 100, 100, 3)