ctheodoris

/

Geneformer

like 297

this addresses the issue when using GeneformerPrecollator if not using numpy array as tensor backend, a ValueError is raise like type of tensor([ 2943, 24469, 12039, ..., 4506, 13856, 10511]) unknown: <class 'torch.Tensor'>. Should be one of a python, numpy, pytorch or tensorflow object.. I believe this is a bug, as I am clearly using a pytorch tensor. And in huggingface transformers library, similar code wrote here