Update README.md

README.md

@@ -20,7 +20,7 @@ tags:
 - code
 - granite
 model-index:
-- name: granite-20b-code-instruct
+- name: granite-20b-code-instruct-r1.1
   results:
   - task:
       type: text-generation
@@ -206,7 +206,7 @@ model-index:
 
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/62cd5057674cdb524450093d/1hzxoPwqkBJXshKVVe6_9.png)
 
-# Granite-20B-Code-Instruct
+# Granite-20B-Code-Instruct-r1.1
 
 ## Model Summary
 **Granite-20B-Code-Instruct-r1.1** is a 20B parameter model fine tuned from *Granite-20B-Code-Instruct-r1.1* on a combination of **permissively licensed** instruction data to enhance instruction following capabilities including mathematical reasoning and problem-solving skills.
@@ -230,7 +230,7 @@ This is a simple example of how to use **Granite-20B-Code-Instruct-r1.1** model.
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # or "cpu"
-model_path = "ibm-granite/granite-20b-code-instruct"
+model_path = "ibm-granite/granite-20b-code-instruct-r1.1"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 # drop device_map if running on CPU
 model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)