Update lib/graph_extract.py

lib/graph_extract.py

@@ -6,17 +6,17 @@ import warnings
 import spaces
 
 flash_attn_installed = False
-try:
-    import subprocess
-    print("Installing flash-attn...")
-    subprocess.run(
-    "pip install flash-attn --no-build-isolation",
-    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
-    shell=True,
-    )
-    flash_attn_installed = True
-except Exception as e:
-    print(f"Error installing flash-attn: {e}")
+# try:
+#     import subprocess
+#     print("Installing flash-attn...")
+#     subprocess.run(
+#     "pip install flash-attn --no-build-isolation",
+#     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+#     shell=True,
+#     )
+#     flash_attn_installed = True
+# except Exception as e:
+#     print(f"Error installing flash-attn: {e}")
 
 
 # Suppress specific warnings