MahmoudAshraf commited on
Commit
afb086c
·
verified ·
1 Parent(s): 85707c9

update for latest version of `ctc-forced-aligner`

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -200,7 +200,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
200
  batch_size = 16
201
 
202
 
203
- alignment_model, alignment_tokenizer, alignment_dictionary = load_alignment_model(
204
  device,
205
  dtype=torch.float16 if device == "cuda" else torch.float32,
206
  )
@@ -225,7 +225,7 @@ tokens_starred, text_starred = preprocess_text(
225
  segments, scores, blank_id = get_alignments(
226
  emissions,
227
  tokens_starred,
228
- alignment_dictionary,
229
  )
230
 
231
  spans = get_spans(tokens_starred, segments, alignment_tokenizer.decode(blank_id))
 
200
  batch_size = 16
201
 
202
 
203
+ alignment_model, alignment_tokenizer = load_alignment_model(
204
  device,
205
  dtype=torch.float16 if device == "cuda" else torch.float32,
206
  )
 
225
  segments, scores, blank_id = get_alignments(
226
  emissions,
227
  tokens_starred,
228
+ alignment_tokenizer,
229
  )
230
 
231
  spans = get_spans(tokens_starred, segments, alignment_tokenizer.decode(blank_id))