Update README.md

README.md

@@ -66,7 +66,7 @@ from transformers import AutoProcessor, AutoModelForImageTextToText
 from accelerate import Accelerator
 
 torch_device = Accelerator().device
-model_checkpoint = "galaxyMindAiLabs/IoGPT-A1"
+model_checkpoint = "galaxyMindAiLabs/IoGPT-A1-Instruct"
 processor = AutoProcessor.from_pretrained(model_checkpoint)
 model = AutoModelForImageTextToText.from_pretrained(model_checkpoint,torch_dtype=torch.bfloat16,  device_map="auto")
 user_prompt = "Why sky is blue?"
@@ -88,7 +88,7 @@ Since this model was trained with Unsloth, using their library provides 2x faste
 from unsloth import FastLanguageModel
 
 model, tokenizer = FastLanguageModel.from_pretrained(
-    model_name = "galaxyMindAiLabs/IoGPT-A1",
+    model_name = "galaxyMindAiLabs/IoGPT-A1-Instruct",
     max_seq_length = 2048,
     dtype = None,
     load_in_4bit = True,
@@ -115,6 +115,7 @@ Framework Versions
  * Pytorch: 2.10.0
  * Datasets: 4.3.0
  * Tokenizers: 0.22.2
+---
 **Citations**
 ```
 @misc{vonwerra2022trl,