POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Usunięcie niepotrzebnego print'a

This commit is contained in:
l.gabrysiak 2025-02-25 17:45:01 +01:00
parent becb46b4ba
commit a1978e7683
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hft.py
View File

@ -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