From 2bce122ddf9245128788426d18e59376f4c3c4e1 Mon Sep 17 00:00:00 2001 From: Tetiana Mohorian Date: Tue, 20 May 2025 21:36:24 +0000 Subject: [PATCH] Aktualizovat peft/model_peft.py --- peft/model_peft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/peft/model_peft.py b/peft/model_peft.py index ae228ed..a11fe98 100644 --- a/peft/model_peft.py +++ b/peft/model_peft.py @@ -29,7 +29,7 @@ val_dataset = create_stratified_split(label_0, label_1, n_samples=10, seed=43) test_dataset = create_stratified_split(label_0, label_1, n_samples=50, seed=44) # Load tokenizer and base model -model_name = "ApoTro/slovak-t5-small" +model_name = "TUKE-KEMT/slovak-t5-base" tokenizer = AutoTokenizer.from_pretrained(model_name, force_download=True) model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2)