modify version

setup.py

@@ -106,7 +106,7 @@ def get_sha(pytorch_root: Union[str, Path]) -> str:
 
 
 def get_version_add(sha: Optional[str] = None) -> str:
-    add_version_path = "jax/version.py"
+    add_version_path = "version.py"
     if sha != 'Unknown':
         if sha is None:
             sha_path = os.getenv('FASTFOLD_DOWNLOAD_PATH', "")
@@ -118,8 +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.21'+'+{}'\n".format(version))
-        file.write("_minimum_jaxlib_version='0.1.69'\n")
+        file.write("__version__='0.2.0'+'+{}'\n".format(version))
     file.close()
 
 
@@ -145,5 +144,5 @@ setup(
     ext_modules=ext_modules,
     package_data={'fastfold': ['model/fastnn/kernel/cuda_native/csrc/*']},
     cmdclass={'build_ext': BuildExtension} if ext_modules else {},
-    #install_requires=['einops', 'colossalai'],
+    install_requires=['einops', 'colossalai'],
 )