Spaces:
Sleeping
Sleeping
Commit
·
992173c
1
Parent(s):
2a3d663
Update app.py
Browse files
app.py
CHANGED
@@ -8,11 +8,12 @@ import datasets
|
|
8 |
# Load your fine-tuned model and tokenizer
|
9 |
model_name = "google/flan-t5-large"
|
10 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
11 |
-
|
12 |
-
|
13 |
-
model.
|
|
|
14 |
|
15 |
-
peft_name = "
|
16 |
peft_config = PeftConfig.from_pretrained(peft_name)
|
17 |
paraphrase_model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
18 |
paraphrase_model = PeftModel.from_pretrained(paraphrase_model, peft_name)
|
|
|
8 |
# Load your fine-tuned model and tokenizer
|
9 |
model_name = "google/flan-t5-large"
|
10 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
11 |
+
peft_name = "checkiejan/flan-t5-prefix-25-9-2"
|
12 |
+
peft_config = PeftConfig.from_pretrained(peft_name)
|
13 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(peft_config.base_model_name_or_path)
|
14 |
+
model = PeftModel.from_pretrained(model, peft_name)
|
15 |
|
16 |
+
peft_name = "checkiejan/prefix-paraphase-50-19-auto"
|
17 |
peft_config = PeftConfig.from_pretrained(peft_name)
|
18 |
paraphrase_model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
19 |
paraphrase_model = PeftModel.from_pretrained(paraphrase_model, peft_name)
|