ClefChen commited on
Commit
286daf8
·
verified ·
1 Parent(s): cb9e318

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ import torch
16
  from transformers import AutoModel, AutoTokenizer
17
 
18
  # 导入预训练模型和分词器
19
- model_name = "microsoft/phi2-base"
20
  tokenizer = AutoTokenizer.from_pretrained(model_name)
21
  model = AutoModel.from_pretrained(model_name)
22
 
 
16
  from transformers import AutoModel, AutoTokenizer
17
 
18
  # 导入预训练模型和分词器
19
+ model_name = "microsoft/phi-2"
20
  tokenizer = AutoTokenizer.from_pretrained(model_name)
21
  model = AutoModel.from_pretrained(model_name)
22