update for latest version of `ctc-forced-aligner`
Browse files
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
|
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 |
-
|
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))
|