diff options
| -rw-r--r-- | .gitignore | 4 | ||||
| -rw-r--r-- | main.py | 4 | ||||
| -rw-r--r-- | pyproject.toml | 14 |
3 files changed, 22 insertions, 0 deletions
@@ -158,3 +158,7 @@ cython_debug/ # and can be added to the global gitignore or merged into this file. For a more nuclear # option (not recommended) you can uncomment the following to ignore the entire idea folder. #.idea/ + +# uv +.python-version +uv.lock
\ No newline at end of file @@ -0,0 +1,4 @@ +from transformers import BertTokenizer, BertModel + +tokenizer = BertTokenizer.from_pretrained("hfl/minirbt-h256") +model = BertModel.from_pretrained("hfl/minirbt-h256")
\ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..f6fb85a --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,14 @@ +[project] +name = "base-model" +version = "0.1.0" +description = "Add your description here" +readme = "README.md" +requires-python = ">=3.12" +dependencies = [ + "torch>=2.9.1", + "transformers>=4.57.3", +] + +[[tool.uv.index]] +url = "https://mirrors.aliyun.com/pypi/simple" +default = true |