check version and modify alphafold.py

fastfold/model/hub/alphafold.py

@@ -427,7 +427,9 @@ class AlphaFold(nn.Module):
             z_prev = z
 
             # [*, N, 3]
-            x_prev = outputs["final_atom_positions"]
+            x_prev = atom14_to_atom37(
+                outputs_sm["positions"][-1], feats
+            )
         
         if no_iter != 3:
             return None,  m_1_prev, z_prev, x_prev

setup.py

@@ -118,7 +118,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
         version += "." + version_dtk
 
     with open(add_version_path, encoding="utf-8", mode="w") as file:
-        file.write("__version__='0.2.0'+'+{}'\n".format(version))
+        file.write("__version__='0.2.1'+'+{}'\n".format(version))
     file.close()
 
 
@@ -132,7 +132,6 @@ def get_version():
 
 setup(
     name='fastfold',
-    #version='0.2.0',
     version=get_version(),
     packages=find_packages(exclude=(
         'assets',

version.py

-__version__ = "0.2.0"
+__version__ = "0.2.1"
\ No newline at end of file