Update README.md

README.md

@@ -6,7 +6,6 @@ datasets:
 language:
 - en
 library_name: transformers
-pipeline_tag: text-generation
 ---
 # Model Card for Alpaca Cerebras-6.7B LoRA
 
@@ -51,6 +50,7 @@ See [github.com/bjoernpl/cerebras-lora](https://github.com/bjoernpl/cerebras-lor
 This model can be easily loaded using the AutoModelForCausalLM functionality:
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from peft import PeftModel
 tokenizer = AutoTokenizer.from_pretrained("cerebras/Cerebras-GPT-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')