Update README.md

README.md

@@ -18,10 +18,10 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 ### Direct Use
 
 ```python
-model = AutoModel.from_pretrained('FrankL/storytellerLM-v0', trust_remote_code=True, torch_dtype=torch.float16)
+model = AutoModel.from_pretrained('FrankL/storytellerLM-v0.1', trust_remote_code=True, torch_dtype=torch.float16)
 model = model.to(device='cuda')
 
-tokenizer = AutoTokenizer.from_pretrained('FrankL/storytellerLM-v0', trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained('FrankL/storytellerLM-v0.1', trust_remote_code=True)
 def inference(
     model: AutoModelForCausalLM,
     tokenizer: AutoTokenizer,