Update README.md

README.md

@@ -163,7 +163,9 @@ We compare our solution with both open automatic spell checkers and the ChatGPT
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 tokenizer = AutoTokenizer.from_pretrained("ai-forever/sage-fredt5-distilled-95m")
-model = AutoModelForSeq2SeqLM.from_pretrained("ai-forever/sage-fredt5-distilled-95m", device_map='cuda')
+model = AutoModelForSeq2SeqLM.from_pretrained("ai-forever/sage-fredt5-distilled-95m")
+
+model.to("cuda")
 
 sentence = "И не чсно прохожим в этот день непогожйи почему я веселый такйо"
 inputs = tokenizer(sentence, max_length=None, padding="longest", truncation=False, return_tensors="pt")