From 84f8adfdbdd96a5bf27c4628247c0ee552a65c87 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Tue, 25 Feb 2025 14:03:55 +0100 Subject: [PATCH] zmiana modelu --- hft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hft.py b/hft.py index fa94e42..cc97948 100644 --- a/hft.py +++ b/hft.py @@ -151,7 +151,7 @@ class CustomTrainer(Trainer): # Inicjalizacja komponentów source_mapper = SourceMapper() -model_name = "perplexity-ai/r1-1776"#"google/gemma-2-2b" +model_name = "Sao10K/Fimbulvetr-11B-v2-GGUF" #"google/gemma-2-2b" tokenizer = AutoTokenizer.from_pretrained(model_name) tokenizer.pad_token = tokenizer.eos_token