fix

app.py

@@ -6,11 +6,14 @@ import subprocess
 
 import llava.serve.gradio_web_server as gws
 
+# Execute the pip install command with additional options
+subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U'])
+
 
 def start_controller():
     print("Starting the controller")
     controller_command = [
-        "python",
+        sys.executable,
         "-m",
         "llava.serve.controller",
         "--host",
@@ -29,7 +32,7 @@ def start_worker(model_path: str, bits=16):
     if bits != 16:
         model_name += f"-{bits}bit"
     worker_command = [
-        "python",
+        sys.executable,
         "-m",
         "llava.serve.model_worker",
         "--host",

requirements.txt

@@ -1,2 +1 @@
-llava-torch
-flash-attn
+llava-torch