kule2340 commited on
Commit
303bd34
1 Parent(s): ea2e28e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -77,12 +77,12 @@ article_en = "The head of the United Nations says there is no military solution
77
  model = MBartForConditionalGeneration.from_pretrained("SnypzZz/Llama2-13b-Language-translate")
78
  tokenizer = MBart50TokenizerFast.from_pretrained("SnypzZz/Llama2-13b-Language-translate", src_lang="en_XX")
79
 
80
- model_inputs = tokenizer(article_en, return_tensors="pt")
81
 
82
  # translate from English to Hindi
83
  generated_tokens = model.generate(
84
  **model_inputs,
85
- forced_bos_token_id=tokenizer.lang_code_to_id["hi_IN"]
86
  )
87
  tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
88
  # => 'संयुक्त राष्ट्र के नेता कहते हैं कि सीरिया में कोई सैन्य समाधान नहीं है'
 
77
  model = MBartForConditionalGeneration.from_pretrained("SnypzZz/Llama2-13b-Language-translate")
78
  tokenizer = MBart50TokenizerFast.from_pretrained("SnypzZz/Llama2-13b-Language-translate", src_lang="en_XX")
79
 
80
+ model_inputs = tokenizer(article_en, return_tensors="de")
81
 
82
  # translate from English to Hindi
83
  generated_tokens = model.generate(
84
  **model_inputs,
85
+ forced_bos_token_id=tokenizer.lang_code_to_id["de_DE"]
86
  )
87
  tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
88
  # => 'संयुक्त राष्ट्र के नेता कहते हैं कि सीरिया में कोई सैन्य समाधान नहीं है'