From 37d386ca0badb3c9c203cdce66f81cf525a4ae14 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Tue, 25 Feb 2025 18:08:57 +0100 Subject: [PATCH] GPU --- hft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hft.py b/hft.py index 2877c67..6c66b01 100644 --- a/hft.py +++ b/hft.py @@ -162,7 +162,7 @@ tokenized_dataset = dataset.map(tokenize_function, batched=True, batch_size=8) config = AutoModelForCausalLM.from_pretrained(model_name).config #print("Vocabulary size:", config.vocab_size) model = CustomModel(model_name, config) -model.to("cpu") # Zmienione na CPU dla debugowania +#model.to("cpu") # Zmienione na CPU dla debugowania # Konfiguracja treningu training_args = TrainingArguments(