I am now getting this error :
0%|          | 0/1 [00:00<?, ?ba/s]
100%|██████████| 1/1 [00:00<00:00,  5.38ba/s]
100%|██████████| 1/1 [00:00<00:00,  5.38ba/s]
Traceback (most recent call last):
File “/gpfs7kw/linkhome/rech/genlig01/umg16uw/test/expe_5/platal/exec.py”, line 10, in 
classification_fmc(model, file)
File “/gpfs7kw/linkhome/rech/genlig01/umg16uw/test/expe_5/platal/classification_fmc_platal.py”, line 89, in classification_fmc
Xtest_emb, s = get_flaubert_layer(Xtest[‘sent’], path_to_model_lge) # index 2 correspond to sentences
File “/gpfs7kw/linkhome/rech/genlig01/umg16uw/test/expe_5/platal/…/traitements/functions_for_processing.py”, line 212, in get_flaubert_layer
hidden_state = flaubert(**tokenized_dataset[0])
File “/linkhome/rech/genlig01/umg16uw/.conda/envs/bert/lib/python3.9/site-packages/torch/nn/modules/module.py”, line 1051, in _call_impl
return forward_call(*input, **kwargs)
File “/linkhome/rech/genlig01/umg16uw/.conda/envs/bert/lib/python3.9/site-packages/transformers/models/flaubert/modeling_flaubert.py”, line 174, in forward
bs, slen = input_ids.size()
ValueError: not enough values to unpack (expected 2, got 1)
srun: error: r13i1n0: task 0: Exited with exit code 1
srun: Terminating job step 428869.0