yumoqing
769fc4968e
- Wan2.2-TI2V-5B GPU 视频推理 - ahserver + longtasks 异步任务队列 - OpenAI 兼容 API: POST /api/submit, GET /api/task, GET /api/status - 模型常驻内存，惰性加载 - 全局串行推理锁（GPU 安全） - 支持 t2v/i2v/ti2v/s2v 四种任务类型
18 lines
471 B
Plaintext
18 lines
471 B
Plaintext
# -*- coding:utf-8 -*-
|
|
# GET /api/task?task_id=xxx - 查询任务状态
|
|
|
|
import json
|
|
from ahserver.serverenv import ServerEnv
|
|
|
|
task_id = params_kw.get('task_id', '')
|
|
if not task_id:
|
|
return json.dumps({'error': 'task_id is required'}, ensure_ascii=False)
|
|
|
|
env = ServerEnv()
|
|
longtasks = env.longtasks
|
|
if longtasks is None:
|
|
return json.dumps({'error': 'service not ready'}, ensure_ascii=False)
|
|
|
|
status = await longtasks.get_status(task_id)
|
|
return json.dumps(status)
|