hf-internal-testing
/

tiny-random-GraniteForCausalLM

Text Generation

Inference Endpoints

Model card Files Files and versions Community

Xenova HF staff commited on 20 days ago

Commit

c3074eb

•

1 Parent(s): 8952624

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -23,6 +23,7 @@ model = GraniteForCausalLM(GraniteConfig(
     num_hidden_layers=2,
     num_key_value_heads=4,
     pad_token_id=0,
 ))
 tokenizer = AutoTokenizer.from_pretrained('ibm/PowerLM-3b')

     num_hidden_layers=2,
     num_key_value_heads=4,
     pad_token_id=0,
+    vocab_size=49152,
 ))
 tokenizer = AutoTokenizer.from_pretrained('ibm/PowerLM-3b')