Hi, great work! I just noticed a small typo : In the inference section of the readme, the supposedly <model_checkpoint> is written <model_checkpiont>