Update README.md

README.md

@@ -21,8 +21,11 @@ Abstract: The study of biological materials and bio-inspired materials science i
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/623ce1c6b66fedf374859fe7/Xdp_nCYiF2IAPamG5ffIC.png)
 
 ```
-model = PeftModel.from_pretrained('lamm-mit/BioinspiredLLM')
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from accelerate import infer_auto_device_map
+
+model = AutoModelForCausalLM.from_pretrained('lamm-mit/BioinspiredLLM')
+tokenizer = AutoTokenizer.from_pretrained('lamm-mit/BioinspiredLLM')
 ```
 
 Variants of the model are included, featuring various GGUF versions for use withm llama.cpp, for instance. In LM Studio, choose the "ChatML" prompt template.