Update README.md

README.md

@@ -18,16 +18,16 @@ Please note the custom license from [the original](https://huggingface.co/01-ai/
 ```python
 # !pip install -U -q transformers accelerate sentencepiece bitsandbytes
 import torch
-
-# Load model directly
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-tokenizer = AutoTokenizer.from_pretrained("pszemraj/Yi-6B-200K-Llama-sharded", use_fast=False)
+tokenizer = AutoTokenizer.from_pretrained(
+    "pszemraj/Yi-6B-200K-Llama-sharded", use_fast=False
+)
 model = AutoModelForCausalLM.from_pretrained(
     "pszemraj/Yi-6B-200K-Llama-sharded", load_in_4bit=True
 )
 
-prompt = "Custom non-commercial licenses are just so fun, aren't they? The best thing about them is"
+prompt = "Custom non-commercial software licenses are just so fun, aren't they? The best thing about them is"
 
 
 if torch.cuda.is_available():