summaryrefslogtreecommitdiffstatshomepage
path: root/main.py
diff options
context:
space:
mode:
authorHsiangNianian <i@jyunko.cn>2025-12-29 19:59:36 +0800
committerHsiangNianian <i@jyunko.cn>2025-12-29 19:59:36 +0800
commit12910f3a937633a25aa0de463a6edf756f2b8cdd (patch)
tree96d715856d9fbc85dd23525d5596b21ff887f17a /main.py
parent2f4904f6dbd34a492a7783c037db85237588ae06 (diff)
downloadbase-model-12910f3a937633a25aa0de463a6edf756f2b8cdd.tar.gz
base-model-12910f3a937633a25aa0de463a6edf756f2b8cdd.zip
refactor: Refactor code structure for improved readability and maintainability
Diffstat (limited to 'main.py')
-rw-r--r--main.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..de058e7
--- /dev/null
+++ b/main.py
@@ -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