From 298035052b3e3d083b57f5dbac0e86de4f94efba Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Tue, 30 Dec 2025 20:39:34 +0800 Subject: refactor: Update model download functionality and improve inference module to support automatic model retrieval from GitHub releases --- pyproject.toml | 17 +++-------------- 1 file changed, 3 insertions(+), 14 deletions(-) (limited to 'pyproject.toml') diff --git a/pyproject.toml b/pyproject.toml index b92fae3..f9e8216 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -56,22 +56,11 @@ Issues = "https://github.com/HydroRoll-Team/base-model/issues" [tool.hatch.build.targets.wheel] packages = ["src/base_model_trpgner"] -artifacts = [ - "src/base_model_trpgner/**/*.py", - "models/trpg-final/model.onnx", - "models/trpg-final/model.onnx.data", - "models/trpg-final/config.json", - "models/trpg-final/tokenizer.json", - "models/trpg-final/tokenizer_config.json", - "models/trpg-final/special_tokens_map.json", - "models/trpg-final/vocab.txt", -] - -[tool.hatch.build.targets.wheel.shared-data] -"models/trpg-final" = "base_model_trpgner/models/trpg-final" +# 仅包含 Python 源码，模型文件首次运行时自动从 GitHub Release 下载 +artifacts = ["src/base_model_trpgner/**/*.py"] [tool.hatch.build.targets.sdist] -include = ["/src", "/models/trpg-final", "/README.md", "/COPYING"] +include = ["/src", "/README.md", "/COPYING"] [tool.ruff] exclude = [ -- cgit v1.2.3-70-g09d2