Muhammadreza commited on
Commit
94fdb18
·
verified ·
1 Parent(s): b122a8f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -80,7 +80,7 @@ import torch
80
  from transformers import AutoModelForImageTextToText, AutoTokenizer
81
 
82
  model_id = "MaralGPT/MaralGPT-Mythos-9B-2606"
83
- tokenizer = AutoTokenizer.from_pretrained("haghiri")
84
  model = AutoModelForImageTextToText.from_pretrained(
85
  model_id, dtype="bfloat16", device_map="cuda"
86
  )
 
80
  from transformers import AutoModelForImageTextToText, AutoTokenizer
81
 
82
  model_id = "MaralGPT/MaralGPT-Mythos-9B-2606"
83
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
84
  model = AutoModelForImageTextToText.from_pretrained(
85
  model_id, dtype="bfloat16", device_map="cuda"
86
  )