Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -51,13 +51,13 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
51
 
52
 
53
  # Inference Example
54
- ```
55
- python
56
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
57
  import torch
58
 
59
- model = AutoModelForSeq2SeqLM.from_pretrained("fine_tuned_models_fp16/en-mr/final/", torch_dtype=torch.float16).to("cuda")
60
- tokenizer = AutoTokenizer.from_pretrained("fine_tuned_models_fp16/en-mr/final/")
 
61
 
62
  inputs = tokenizer("Hello, how are you?", return_tensors="pt").to("cuda")
63
  outputs = model.generate(**inputs)
 
51
 
52
 
53
  # Inference Example
54
+ ```python
 
55
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
56
  import torch
57
 
58
+ model_name = "AventIQ-AI/opus-mt-en-roa_multilanguageTranslation"
59
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to("cuda")
60
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
61
 
62
  inputs = tokenizer("Hello, how are you?", return_tensors="pt").to("cuda")
63
  outputs = model.generate(**inputs)