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

poprawka trener c.d.

This commit is contained in:
l.gabrysiak 2025-02-25 13:23:14 +01:00
parent 6d1150308b
commit d4f742d0a8
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
hft.py
View File

@ -99,13 +99,14 @@ class CustomModel(AutoModelForCausalLM):
# Dostosowany Trainer
class CustomTrainer(Trainer):
def compute_loss(self, model, inputs, return_outputs=False, num_items_in_batch=None):
def compute_loss(self, model, inputs, return_outputs=False):
labels = inputs.pop("labels")
source = inputs.pop("source")
source = inputs.pop("source", None) # Użyj None jako wartości domyślnej
outputs = model(**inputs, labels=labels)
loss = outputs.loss
return (loss, outputs) if return_outputs else loss
# Przygotowanie modelu i tokenizera
model_name = "google/gemma-2-2b"
tokenizer = AutoTokenizer.from_pretrained(model_name)