Views
No views yet
main1{
2 "attention_probs_dropout_prob": 0.1,
3 "hidden_dropout_prob": 0.1,
4 "hidden_size": 768,
5 "intermediate_size": 2560,
6 "max_position_embeddings": 512,
7 "position_bucket_size": 32,
8 "num_attention_heads": 12,
9 "num_hidden_layers": 12,
10 "vocab_size": 16384,
11 "layer_norm_eps": 1e-05
12}tokenizer_eng.json1from transformers import AutoTokenizer, AutoModelForMaskedLM
2model_id = 'haznitrama/babybabellm-multi_gpu-gpt_bert-eng-main'
3tok = AutoTokenizer.from_pretrained(model_id)
4model = AutoModelForMaskedLM.from_pretrained(model_id, trust_remote_code=True)
5out = model(**tok('Hello world', return_tensors='pt'))trust_remote_code=True due to custom architecture.