fix: incorrect condition control flow
#261
by
tpob
- opened
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
ctheodoris
changed pull request status to
merged
Thank you for the fix!