Update README.md

README.md

@@ -15,12 +15,12 @@ tags: []
 
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-model = AutoModelForCausalLM.from_pretrained("AlexWortega/v5-it",
+model = AutoModelForCausalLM.from_pretrained("Vikhrmodels/Vikhr-7B-instruct_0.4",
                                              device_map="auto",
                                              attn_implementation="flash_attention_2",
                                              torch_dtype=torch.bfloat16)
 
-tokenizer = AutoTokenizer.from_pretrained("AlexWortega/v5-it")
+tokenizer = AutoTokenizer.from_pretrained("Vikhrmodels/Vikhr-7B-instruct_0.4")
 from transformers import  AutoTokenizer, pipeline
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 prompts = [