Update README.md

README.md

@@ -21,10 +21,10 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 model = AutoModelForCausalLM.from_pretrained("Vikhrmodels/it-5.2-fp16-cp",
                                              device_map="auto",
-                                             attn_implementation="flash_attention_2",
+                                             attn_implementation="sdpa",
                                              torch_dtype=torch.bfloat16)
 
-tokenizer = AutoTokenizer.from_pretrained("Vikhrmodels/Vikhr-7B-instruct_0.4")
+tokenizer = AutoTokenizer.from_pretrained("Vikhrmodels/it-5.2-fp16-cp")
 from transformers import  AutoTokenizer, pipeline
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 prompts = [