From 5aef37828d6ae3fa61184fd6910c805bbabd797a Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Tue, 25 Feb 2025 17:58:35 +0100 Subject: [PATCH] =?UTF-8?q?Usuni=C4=99cie=20niepotrzebnego=20print'a?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- hft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hft.py b/hft.py index 86a3048..2ae6488 100644 --- a/hft.py +++ b/hft.py @@ -160,7 +160,7 @@ tokenized_dataset = dataset.map(tokenize_function, batched=True, batch_size=8) # Inicjalizacja modelu config = AutoModelForCausalLM.from_pretrained(model_name).config -print("Vocabulary size:", config.vocab_size) +#print("Vocabulary size:", config.vocab_size) model = CustomModel(model_name, config) model.to("cpu") # Zmienione na CPU dla debugowania