jiachenl commited on
Commit
2e824a9
β€’
1 Parent(s): c3f3b0b
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import cumo.serve.gradio_web_server as gws
11
  #os.system("pip install flash-attn --no-build-isolation")
12
 
13
  # Execute the pip install command with additional options
14
- subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U'])
15
 
16
  def start_controller():
17
  print("Starting the controller")
@@ -45,7 +45,6 @@ def start_worker(model_path: str, bits=16):
45
  model_path,
46
  "--model-name",
47
  model_name,
48
- "--use-flash-attn",
49
  ]
50
  if bits != 16:
51
  worker_command += [f"--load-{bits}bit"]
 
11
  #os.system("pip install flash-attn --no-build-isolation")
12
 
13
  # Execute the pip install command with additional options
14
+ #subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation', '-U'])
15
 
16
  def start_controller():
17
  print("Starting the controller")
 
45
  model_path,
46
  "--model-name",
47
  model_name,
 
48
  ]
49
  if bits != 16:
50
  worker_command += [f"--load-{bits}bit"]