`batch_size, seq_length = input_ids.shape ValueError: not enough values to unpack (expected 2, got 1)`

rmihaylov / falcontune

Tune any FALCON in 4-bit

Apache License 2.0

468 stars 51 forks source link

Open MohamedAliRashad opened 1 year ago

MohamedAliRashad commented 1 year ago

Why i am getting this error ?

yxli2123 commented 1 year ago

I think your input is a single sentence. Even if your input is a single sentence, you should also make it a batch.