Solving https://github.com/instadeepai/nucleotide-transformer/issues/137 and https://github.com/instadeepai/nucleotide-transformer/issues/138
Solving #137 and #138