Update README.md

README.md

@@ -32,16 +32,14 @@ You can find more details in the [technical report](https://arxiv.org/abs/2310.1
 
 ## Use in Transformers
 ## Load model directly
-import torch
+import torch \
 from transformers import AutoTokenizer, AutoModelForCausalLM, MistralForCausalLM
 
 model = MistralForCausalLM.from_pretrained("ai-agi/neural-zephyr", use_cache=False,  torch_dtype=torch.bfloat16, device_map="auto") \
 state_dict = torch.load('model_weights.pth') \
-model.load_state_dict(state_dict) \
+model.load_state_dict(state_dict) 
 
-tokenizer = AutoTokenizer.from_pretrained("ai-agi/neural-zephyr", use_fast=True)
-
-if tokenizer.pad_token is None:
-
-    tokenizer.pad_token = tokenizer.eos_token)
+tokenizer = AutoTokenizer.from_pretrained("ai-agi/neural-zephyr", use_fast=True) \
+if tokenizer.pad_token is None: \
+    tokenizer.pad_token = tokenizer.eos_token)