From c1df27a47266a826e3aed3eda445358c5801d767 Mon Sep 17 00:00:00 2001 From: wanghh2000 Date: Fri, 29 Dec 2023 23:00:21 +0800 Subject: [PATCH] test flan-t5 --- finetune/finetune-t5-lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune/finetune-t5-lora.py b/finetune/finetune-t5-lora.py index f8ecb34..851735d 100644 --- a/finetune/finetune-t5-lora.py +++ b/finetune/finetune-t5-lora.py @@ -3,7 +3,7 @@ import torch device = torch.device("cpu") 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" print('********* before finetune ***********') from transformers import AutoTokenizer, AutoModelForSeq2SeqLM