POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mod

This commit is contained in:
l.gabrysiak 2025-02-25 15:24:49 +01:00
parent 4014b12ab4
commit c42a4f6f9a
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
hft.py
View File

@ -12,9 +12,6 @@ import json
from collections import defaultdict
from huggingface_hub import login
import torch
torch.cuda.empty_cache()
login(token="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")
os.environ["TOKENIZERS_PARALLELISM"] = "false"
@ -164,7 +161,7 @@ tokenized_dataset = dataset.map(tokenize_function, batched=True, batch_size=8)
# Inicjalizacja modelu
config = AutoModelForCausalLM.from_pretrained(model_name).config
model = CustomModel.from_pretrained(model_name, config=config)
model.gradient_checkpointing_enable()
model.to("cpu")
# Konfiguracja treningu
training_args = TrainingArguments(
@ -177,8 +174,7 @@ training_args = TrainingArguments(
logging_steps=100,
save_strategy="steps",
save_steps=1000,
report_to="none",
gradient_checkpointing=True
report_to="none"
)
# Trening