合并拉取请求 #44

fix #43

Author: mrhan1993

apis/utils/post_worker.py

@@ -3,6 +3,7 @@
 """
 import datetime
 import os
+import numpy as np
 
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
@@ -48,7 +49,8 @@ async def post_worker(task: AsyncTask, started_at: int, target_name: str, ext: s
     if task.last_stop in ['stop', 'skip']:
         task_status = task.last_stop
     try:
-        results = change_filename(task.results, target_name, ext)
+        task_results = [task_result for task_result in task.results if not isinstance(task_result, np.ndarray)]
+        results = change_filename(task_results, target_name, ext)
         task.results = results
         query = session.query(GenerateRecord).filter(GenerateRecord.task_id == task.task_id).first()
         query.start_mills = started_at
@@ -58,8 +60,9 @@ async def post_worker(task: AsyncTask, started_at: int, target_name: str, ext: s
         query.result = url_path(results)
         finally_result = str(query)
         session.commit()
-        await send_result_to_web_hook(query.webhook_url, finally_result)
+        await send_result_to_web_hook(str(query.webhook_url), finally_result)
         return finally_result
     except Exception as e:
         print(e)
+
     CurrentTask.ct = None

docker.md

@@ -9,7 +9,7 @@ The docker image is based on NVIDIA CUDA 12.4 and PyTorch 2.1, see [Dockerfile](
 
 ## Quick start
 
-**More information in the [notes](#notes).**
+**More information in the [notes](#Notes).**
 
 ### Running with Docker Compose
 
@@ -98,13 +98,13 @@ Since `/content/data` is a persistent volume folder, your files will be persiste
 
 ### Paths inside the container
 
-|Path|Details|
-|-|-|
-|/content/app|The application stored folder|
-|/content/app/models.org|Original 'models' folder.<br> Files are copied to the '/content/app/models' which is symlinked to '/content/data/models' every time the container boots. (Existing files will not be overwritten.) |
-|/content/data|Persistent volume mount point|
-|/content/data/models|The folder is symlinked to '/content/app/models'|
-|/content/data/outputs|The folder is symlinked to '/content/app/outputs'|
+| Path                    | Details                                                                                                                                                                                            |
+|-------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| /content/app            | The application stored folder                                                                                                                                                                      |
+| /content/app/models.org | Original 'models' folder.<br> Files are copied to the '/content/app/models' which is symlinked to '/content/data/models' every time the container boots. (Existing files will not be overwritten.) |
+| /content/data           | Persistent volume mount point                                                                                                                                                                      |
+| /content/data/models    | The folder is symlinked to '/content/app/models'                                                                                                                                                   |
+| /content/data/outputs   | The folder is symlinked to '/content/app/outputs'                                                                                                                                                  |
 
 ### Environments