Update README.md

README.md

@@ -115,11 +115,11 @@ config = ModelConfig(vocab_size=32000, hidden_size=768, n_heads=12,
                      dropout=0.0, pre_norm=True, tie_weights=True)
 
 model = Transformer(config)
-model.load_state_dict(torch.load('LumenBase.safetensors'))
+model.load_state_dict(torch.load('model.safetensors'))
 model.eval()
 
 # Generate text
-tokenizer = Tokenizer.from_file('LumenTokenizer.json')
+tokenizer = Tokenizer.from_file('tokenizer.json')
 prompt = "Once upon a time"
 input_ids = torch.tensor([tokenizer.encode(prompt).ids])