Spaces:
Running
on
Zero
Running
on
Zero
jiachenl
commited on
Commit
·
2e824a9
1
Parent(s):
c3f3b0b
update
Browse files
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"]
|