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

gemma-2-2b

This commit is contained in:
l.gabrysiak 2025-02-25 10:40:49 +01:00
parent 29033583b4
commit 7d7578833e
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

@ -100,7 +100,7 @@ class CustomTrainer(Trainer):
return (loss, outputs) if return_outputs else loss return (loss, outputs) if return_outputs else loss
# Przygotowanie modelu i tokenizera # Przygotowanie modelu i tokenizera
model_name = "google/gemma-2b" model_name = "google/gemma-2-2b"
tokenizer = AutoTokenizer.from_pretrained(model_name) tokenizer = AutoTokenizer.from_pretrained(model_name)
model = CustomModel.from_pretrained(model_name) model = CustomModel.from_pretrained(model_name)