Compare commits
12 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b799ebbbee | ||
|
|
24928f5678 | ||
|
|
3eed33f0d8 | ||
|
|
a2ef82d954 | ||
|
|
2198aed3c2 | ||
|
|
965342bafe | ||
|
|
5bb32e6bdb | ||
|
|
75ba3bb301 | ||
|
|
e28f2e3eca | ||
|
|
e0b6c3b5a8 | ||
|
|
f438206c0e | ||
|
|
12616c3816 |
@@ -4,6 +4,8 @@ ENV PYTHONUNBUFFERED=1
|
|||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN mkdir -p /app/data
|
||||||
|
|
||||||
COPY requirements.txt .
|
COPY requirements.txt .
|
||||||
RUN pip install --no-cache-dir -r requirements.txt
|
RUN pip install --no-cache-dir -r requirements.txt
|
||||||
|
|
||||||
|
|||||||
7
main.py
7
main.py
@@ -56,7 +56,7 @@ def cancel_all_tasks():
|
|||||||
|
|
||||||
log("=" * 50)
|
log("=" * 50)
|
||||||
log("开始取消所有任务")
|
log("开始取消所有任务")
|
||||||
log(f"目标GPU: {TARGET_GPU}")
|
log(f"目标: 所有GPU")
|
||||||
|
|
||||||
# 1. 查询所有任务
|
# 1. 查询所有任务
|
||||||
all_tasks = []
|
all_tasks = []
|
||||||
@@ -66,8 +66,7 @@ def cancel_all_tasks():
|
|||||||
resp = requests.get(
|
resp = requests.get(
|
||||||
f"{MODELHUB_API}/adapt/task/page",
|
f"{MODELHUB_API}/adapt/task/page",
|
||||||
headers={'Xc-Token': TARGET_TOKEN, 'Accept': 'application/json'},
|
headers={'Xc-Token': TARGET_TOKEN, 'Accept': 'application/json'},
|
||||||
params={'current': page, 'pageSize': 100, 'onlyMine': 'true',
|
params={'current': page, 'pageSize': 100, 'onlyMine': 'true'},
|
||||||
'gpuType': TARGET_GPU},
|
|
||||||
timeout=15
|
timeout=15
|
||||||
)
|
)
|
||||||
data = resp.json()
|
data = resp.json()
|
||||||
@@ -172,7 +171,7 @@ def main():
|
|||||||
server.timeout = 1
|
server.timeout = 1
|
||||||
|
|
||||||
log(f"Cancel-All 智能体启动 | {HOST}:{PORT}")
|
log(f"Cancel-All 智能体启动 | {HOST}:{PORT}")
|
||||||
log(f"目标GPU: {TARGET_GPU}")
|
log(f"目标: 所有GPU")
|
||||||
|
|
||||||
# 启动后自动取消所有任务
|
# 启动后自动取消所有任务
|
||||||
def _auto_cancel():
|
def _auto_cancel():
|
||||||
|
|||||||
Reference in New Issue
Block a user