Update README.md

README.md

@@ -44,11 +44,14 @@ https://huggingface.co/cosimoiaia/Loquace-20B   -   Based on gpt-neox-20B
 
 
 ```python
-from peft import PeftModel
-from transformers import LLaMATokenizer, LLaMAForCausalLM, GenerationConfig
+from transformers import (
+    AutoTokenizer,
+    AutoModelForCausalLM,
+    BitsAndBytesConfig
+)
 
-tokenizer = LLaMATokenizer.from_pretrained("cosimoiaia/Loquace-70m")
-model = LLaMAForCausalLM.from_pretrained(
+tokenizer = AutoTokenizer.from_pretrained("cosimoiaia/Loquace-70m", padding_side="right", use_fast=True)
+model = AutoModelForCausalLM.from_pretrained(
     "cosimoiaia/Loquace-70m",
     load_in_8bit=True,
     device_map="auto",