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

inference.py

@@ -411,7 +411,7 @@ def inference_monomer_model(args):
     batch = processed_feature_dict
 
     manager = mp.Manager()
-    result_q = manager.Queue()
+    result_q = manager.Queue()                         
     torch.multiprocessing.spawn(inference_model, nprocs=args.gpus, args=(args.gpus, result_q, batch, args))
 
     out = result_q.get()

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"
\ No newline at end of file
+__version__ = "0.2.1"
\ No newline at end of file