github
/
ChatGLM2-6B
mirror of https://github.com/THUDM/ChatGLM2-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

test flan-t5

pull/672/head
wanghh2000 2023-12-29 23:00:21 +08:00
parent b0edfbcbe3
commit c1df27a472
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/finetune-t5-lora.py
View File

@ -3,7 +3,7 @@ import torch
device = torch.device("cpu") device = torch.device("cpu")
checkpoint = "/Users/hhwang/models/t5-small" checkpoint = "/Users/hhwang/models/t5-small"
# checkpoint = "/Users/hhwang/models/flan-t5-small" checkpoint = "/Users/hhwang/models/flan-t5-small"
prompt = "translate English to German: That is good" prompt = "translate English to German: That is good"
print('********* before finetune ***********') print('********* before finetune ***********')
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM from transformers import AutoTokenizer, AutoModelForSeq2SeqLM