Skip to content

Commit 773f56f

Browse files
committed
fix: gpus list
1 parent 0f9dff2 commit 773f56f

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

nebula/frontend/app.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1231,14 +1231,12 @@ async def assign_available_gpu(scenario_data, role):
12311231
if scenario_data["accelerator"] == "cpu":
12321232
scenario_data["gpu_id"] = []
12331233
else:
1234-
available_gpus = await get_available_gpus()
1235-
1234+
response = await get_available_gpus()
1235+
available_gpus = response.get("available_gpus")
12361236
if role == "user":
1237-
json_available_gpus = available_gpus.pop()
1238-
scenario_data["gpu_id"] = json_available_gpus
1237+
scenario_data["gpu_id"] = available_gpus.pop()
12391238
elif role == "admin":
1240-
json_available_gpus = available_gpus
1241-
scenario_data["gpu_id"] = json_available_gpus
1239+
scenario_data["gpu_id"] = available_gpus
12421240
else:
12431241
scenario_data["gpu_id"] = []
12441242

0 commit comments

Comments
 (0)