A NOVA Finetuned model which is specifically trained for decision-driven Story generator.
1{
2 "model_type": "nova",
3 "vocab_size": 6000,
4 "block_size": 256,
5 "n_embd": 640,
6 "n_layer": 4,
7 "n_head": 8
8}
1git clone https://huggingface.co/harshit36/Nova-Verse
2cd Nova-Verse
1import sys
2sys.path.append("./Nova-Verse/") # add current dir to path
3
4from transformers import PreTrainedTokenizerFast
5from nova_modelling import NovaConfig, NovaForCausalLM
6
7# Load tokenizer
8tokenizer = PreTrainedTokenizerFast.from_pretrained("harshit36/Nova-Verse")
9
10# Load config
11config = NovaConfig.from_pretrained("harshit36/Nova-Verse")
12
13# Instantiate model using your custom class
14model = NovaForCausalLM(config)
15model = model.from_pretrained("harshit36/Nova-Verse")
16
17# Use the model
18input_ids = tokenizer("Hello world", return_tensors="pt").input_ids
19output = model.generate(input_ids)
20print(tokenizer.decode(output[0], skip_special_tokens=True).replace(" ","").replace("Ġ"," ").replace("Ċ","\n"))
21
Hybrid Positional Encoding Research model (Combination of Sinusoidal and learnable encodings)