fix: main_server 支持多 GPU 适配，修复 #11

Dramwig · Dramwig · commit a6eccc2361ce · 2025-09-09T20:37:10.000+08:00
diff --git a/main_cli.py b/main_cli.py
@@ -3,7 +3,6 @@
 from flowline import run_cli
 
 def func(dict, gpu_id, sorted_gpu_ids):
-    print(sorted_gpu_ids)
     return "CUDA_VISIBLE_DEVICES="+str(gpu_id)+" python -u test/test.py "+ " ".join([f"--{k} {v}" for k, v in dict.items()])
 
 def cmp(info1, info2):
diff --git a/main_server.py b/main_server.py
@@ -3,7 +3,7 @@
 from flowline import get_app
 from flask_cors import CORS
 
-def func(dict, gpu_id):
+def func(dict, gpu_id, sorted_gpu_ids):
     return "CUDA_VISIBLE_DEVICES="+str(gpu_id)+" python -u test/test.py "+ " ".join([f"--{k}={v}" for k, v in dict.items()])
 
 def main():
diff --git a/test/todo.xlsx b/test/todo.xlsx
