Update README.md

README.md

@@ -44,12 +44,15 @@ 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-410m")
-model = LLaMAForCausalLM.from_pretrained(
-    "cosimoiaia/Loquace-410m",
+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",
     quantization_config=BitsAndBytesConfig(