tm216ux/bakalarka_praca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Aktualizovat peft/model_peft.py

Browse Source
This commit is contained in:
Tetiana Mohorian 2025-05-20 21:36:24 +00:00
parent 599ee2f5dc
commit 2bce122ddf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
peft/model_peft.py
View File

@ -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)