From 172a3152b42af36443ec6a0a39969471c35b893d Mon Sep 17 00:00:00 2001
From: maliubiao <maliubiao@gmail.com>
Date: 星期日, 29 十二月 2024 22:33:22 +0800
Subject: [PATCH] 允许 model.generate 使用bytes io, 以便不写入文件,节省io时间 (#2343)
---
setup.py | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/setup.py b/setup.py
index ccdd1f5..21c51ce 100644
--- a/setup.py
+++ b/setup.py
@@ -41,6 +41,7 @@
"tensorboardX",
# "rotary_embedding_torch",
"requests",
+ "modelscope",
],
# train: The modules invoked when training only.
"train": [
--
Gitblit v1.9.1