modify version

setup.py

@@ -113,19 +113,19 @@ else:
                         extra_cuda_flags + cc_flag))
 
 
-def get_sha(pytorch_root: Union[str, Path]) -> str:
+def get_sha(root: Union[str, Path]) -> str:
     try:
-        return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=pytorch_root).decode('ascii').strip()
+        return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=root).decode('ascii').strip()
     except Exception:
         return 'Unknown'
 
 
 def get_version_add(sha: Optional[str] = None) -> str:
+    fastfold_root = os.path.dirname(os.path.abspath(__file__))
     add_version_path = "version.py"
     if sha != 'Unknown':
         if sha is None:
-            sha_path = os.getenv('FASTFOLD_DOWNLOAD_PATH', "")
-            sha = get_sha(sha_path)
+            sha = get_sha(fastfold_root)
         version = 'git' + sha[:7]
 
     if os.getenv('FASTFOLD_BUILD_VERSION'):
@@ -158,5 +158,5 @@ setup(
     ext_modules=ext_modules,
     package_data={'fastfold': ['model/fastnn/kernel/cuda_native/csrc/*', 'common/stereo_chemical_props.txt']},
     cmdclass={'build_ext': BuildExtension} if ext_modules else {},
-    install_requires=['einops', 'colossalai'],
+    install_requires=['einops'],
 )