Downloads · 30 days
3
2% of all-time downloads
babybabellm-gpt_bert-small
haznitrama/babybabellm-gpt_bert-small
babybabellm-gpt_bert-small is a machine learning model from haznitrama. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Rehosted from suchirsalhan/babybabellm-multismall with standardized remote code and automap. - Original modeltype preserved. - Default AutoModel mapping points to GPTBertForCausalLM. - Added both causal & masked LM wr…
All-time downloads
174
Public
Repo size
4.2 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin4.2 GB · 100%
At a glance
- Model type
- gpt_bert
- Access
- Public
- Created
- Sep 22, 2025
- Updated
- Sep 29, 2025
- SHA
- 1610bc52
Model card
From the Hugging Face model README
haznitrama/babybabellm-gpt_bert-small
Rehosted from suchirsalhan/babybabellm-multismall with standardized remote code and auto_map.
- Original
model_typepreserved. - Default AutoModel mapping points to GPTBertForCausalLM.
- Added both causal & masked LM wrappers for evaluation.
Example:
from transformers import AutoTokenizer, AutoModel
m='haznitrama/babybabellm-gpt_bert-small'
tok=AutoTokenizer.from_pretrained(m, trust_remote_code=True)
model=AutoModel.from_pretrained(m, trust_remote_code=True)
print(model(**tok('Hello world', return_tensors='pt')).logits.shape)
Generation:
from transformers import AutoModelForCausalLM, AutoTokenizer
m='haznitrama/babybabellm-gpt_bert-small'
tok=AutoTokenizer.from_pretrained(m, trust_remote_code=True)
model=AutoModelForCausalLM.from_pretrained(m, trust_remote_code=True)
print(tok.decode(model.generate(**tok('Hello', return_tensors='pt'), max_new_tokens=20)[0], skip_special_tokens=True))
- Type
- gpt_bert
- Created
- Sep 22, 2025
- Updated
- Sep 29, 2025