Update README.md

README.md

@@ -8,20 +8,19 @@ library_name: transformers
 pipeline_tag: text-generation
 ---
 
-# open-llama-0.3T-7B-instruction-following-v1
-
+# VMware/open-llama-0.3T-7B-instruct-dolly-hhrlhf
 
 ```
 import os
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = 'VMware/open-llama-0.3T-7B-instruction-following-v1'
+model_name = 'VMware/open-llama-0.3T-7B-instruct-dolly-hhrlhf'
 
 
 tokenizer = AutoTokenizer.from_pretrained(model_name, add_bos_token = True)
 
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype= torch.float16, device_map = 'auto')
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype= torch.float16, device_map = 'sequential')
 
 prompt_template = "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n{instruction}\n\n### Response:"