diff --git a/docs/tutorials/tflite.md b/docs/tutorials/tflite.md index 25f29b0d9..6bb02bd4e 100644 --- a/docs/tutorials/tflite.md +++ b/docs/tutorials/tflite.md @@ -11,7 +11,7 @@ ## Conversion ```bash -python3 examples/train.py \ +python3 examples/tflite.py \ --config-path=/path/to/config.yml.j2 \ --h5=/path/to/weight.h5 \ --bs=1 \ # Batch size diff --git a/examples/inferences/main.py b/examples/inferences/main.py index 393ab20fd..d381c0d99 100644 --- a/examples/inferences/main.py +++ b/examples/inferences/main.py @@ -55,11 +55,8 @@ def main( previous_decoder_states=model.get_initial_decoder_states(), ) ) - print(outputs.tokens) transcript = tokenizer.detokenize(outputs.tokens)[0].numpy().decode("utf-8") - logger.info(f"Transcript: {transcript}") - return transcript if __name__ == "__main__":