From 6d1150308bd9d326fbc1b71dc971e5f2ad48212c Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Tue, 25 Feb 2025 13:19:41 +0100 Subject: [PATCH] trener c.d. --- hft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hft.py b/hft.py index 88ab3b9..f47689c 100644 --- a/hft.py +++ b/hft.py @@ -99,7 +99,7 @@ class CustomModel(AutoModelForCausalLM): # Dostosowany Trainer class CustomTrainer(Trainer): - def compute_loss(self, model, inputs, return_outputs=False): + def compute_loss(self, model, inputs, return_outputs=False, num_items_in_batch=None): labels = inputs.pop("labels") source = inputs.pop("source") outputs = model(**inputs, labels=labels)