last cleanup for the day

lucidrains · Sep 27, 2024 · 72087c3 · 72087c3
1 parent 26c2f9f
commit 72087c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alphafold3_pytorch/alphafold3.py b/alphafold3_pytorch/alphafold3.py
@@ -5307,7 +5307,7 @@ def __init__(
             1.4      # 5 - water
         ])
 
-        self.register_buffer('atom_radii', atom_type_radii)
+        self.register_buffer('atom_radii', atom_type_radii, persistent = False)
 
     @property
     def is_mkdssp_available(self):