Update README.md

README.md

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