Question regarding PyTorch / .safetensors weights for fine-tuning
#2
by johnlockejrr - opened
Hi team, thanks for releasing Indra OCR—the CPU throughput and performance look fantastic!
Right now, the repository only includes .onnx weights, which makes downstream fine-tuning difficult. Do you have any plans to publish the native PyTorch checkpoints or *.safetensors weights for the models so they can be fine-tuned? Would be so cool to have the OCR model, for example, to work on other languages (Hebrew for example, but not only).
Thanks!
Disregard, I converted the weights to pytorch. A friend already finetuned the OCR model.
Very good model! In training same fast as lightning.
For personal use.