mod

2025-02-25 22:03:59 +01:00 · 2025-02-25 22:03:59 +01:00 · 1b4449f619
parent eb69fe1633
commit 1b4449f619
1 changed files with 1 additions and 1 deletions
--- a/hft.py
+++ b/hft.py
@ -245,7 +245,7 @@ def main():
        remove_unused_columns=False
    )

-    trainer = CustomTrainer(
+    trainer = Trainer(
        model=model,
        args=training_args,
        train_dataset=tokenized_dataset,