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

zmiana modelu

This commit is contained in:
l.gabrysiak 2025-02-25 15:04:47 +01:00
parent 204dd4421a
commit 4f8d57e770
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

@ -143,7 +143,7 @@ class CustomModel(GPTNeoForCausalLM):
return outputs
source_mapper = SourceMapper()
model_name = "EleutherAI/gpt-neo-2.7B"
model_name = "EleutherAI/gpt-neo-1.3B"
tokenizer = AutoTokenizer.from_pretrained(model_name)
tokenizer.pad_token = tokenizer.eos_token