超分功能完成
This commit is contained in:
@@ -761,6 +761,20 @@ async def retry_task(
|
||||
raise HTTPException(status_code=400, detail="只有失败任务可以重试")
|
||||
retry_targets = [task]
|
||||
|
||||
upscale_failed_ids = [
|
||||
str(target.id)
|
||||
for target in retry_targets
|
||||
if target.pipeline_stage == ChatGenerationPipelineStage.UPSCALE_FAILED.value
|
||||
]
|
||||
if upscale_failed_ids:
|
||||
raise HTTPException(
|
||||
status_code=409,
|
||||
detail={
|
||||
"message": "画质增强失败任务不能通过普通生成重试,请由管理员使用视频超分恢复命令处理",
|
||||
"task_ids": upscale_failed_ids,
|
||||
},
|
||||
)
|
||||
|
||||
await assert_user_resource_capacity_available(db, current_user.id)
|
||||
enqueue_ids: list[str] = []
|
||||
download_retry_ids: list[str] = []
|
||||
|
||||
Reference in New Issue
Block a user