obscura/minimind
1
0
Fork 0
mirror of https://github.com/jingyaogong/minimind.git synced 2026-11-13 03:01:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

update 0-eval_pretrain.py

Browse Source
This commit is contained in:
gongjy
2024-09-14 14:09:29 +08:00
parent ecf6d44133
commit b00a3b5ff7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
0-eval_pretrain.py
+1 -1
View File
@@ -16,7 +16,7 @@ def count_parameters(model):
def init_model(lm_config):
tokenizer = AutoTokenizer.from_pretrained('./model',
tokenizer = AutoTokenizer.from_pretrained('./model/minimind_tokenizer',
trust_remote_code=True, use_fast=False)
model_from = 1 # 1从权重，2用transformers