Update README.md

README.md

@@ -93,8 +93,8 @@ model_inputs = speech_granite_processor(
 ).to(device)
 
 # The recommended repetition penalty is 3 as long as input IDs are excluded.
-# Otherwise, you should use a reptition penalty of 1 to keep results stable.
-reptition_penalty_processor = RepetitionPenaltyLogitsProcessor(
+# Otherwise, you should use a repetition penalty of 1 to keep results stable.
+repetition_penalty_processor = RepetitionPenaltyLogitsProcessor(
     penalty=3.0,
     prompt_ignore_length=model_inputs["input_ids"].shape[-1],
 )
@@ -108,7 +108,7 @@ model_outputs = speech_granite.generate(
     top_p=1.0,
     length_penalty=1.0,
     temperature=1.0,
-    logits_processor=[reptition_penalty_processor],
+    logits_processor=[repetition_penalty_processor],
     bos_token_id=tokenizer.bos_token_id,
     eos_token_id=tokenizer.eos_token_id,
     pad_token_id=tokenizer.pad_token_id,