bugfix

2024-12-06 14:57:00 +08:00 · 2024-12-06 14:57:00 +08:00 · 6ffe11dbc7
commit 6ffe11dbc7
parent 58a590ca24
2 changed files with 12 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,4 +1,8 @@
 git+https://git.kaiyuancloud.cn/yumoqing/apppublic
 git+https://git.kaiyuancloud.cn/yumoqing/ahserver
+git+https://github.com/huggingface/transformers@21fac7abba2a37fae86106f87fcf9974fd1e3830 
+accelerate
+torch
+torchaudio
 optimum
 auto_gptq
--- a/wwwroot/api/index.dspy
+++ b/wwwroot/api/index.dspy
@ -1,3 +1,9 @@
 info(f'{params_kw=}')
-return await inference(params_kw.prompt, params_kw.image, params_kw.video)
-
+t1 = time.time()
+ret = await inference(params_kw.prompt, params_kw.image, params_kw.video)
+t2 = time.time()
+info(f'{ret=}')
+return {
+	"content":ret
+	"time_cost":t2 - t1
+}