Update README.md

README.md

@@ -52,8 +52,8 @@ This model can be easily loaded using the AutoModelForCausalLM functionality:
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("cerebras/Cerebras-GPT-6.7B")
-model = AutoModelForCausalLM.from_pretrained("cerebras/Cerebras-GPT-6.7B")
-model = PeftModel.from_pretrained(model, "bjoernp/alpaca-cerebras-6.7B")
+model = AutoModelForCausalLM.from_pretrained("cerebras/Cerebras-GPT-6.7B", torch_dtype=torch.float16, device_map='auto', load_in_8bit=True)
+model = PeftModel.from_pretrained(model, "bjoernp/alpaca-cerebras-6.7B", torch_dtype=torch.float16, device_map='auto')
 text = "Generative AI is "
 ```