run with crash

#2
by raycast - opened

use the codet5p-2b model run example crash with log and if i change to codet5p-220m work fine
Traceback (most recent call last):
File "C:\Python39\lib\site-packages\torch\utils_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "C:\Python39\lib\site-packages\transformers\generation\utils.py", line 1329, in generate
model_kwargs = self._prepare_encoder_decoder_kwargs_for_generation(
File "C:\Python39\lib\site-packages\transformers\generation\utils.py", line 642, in _prepare_encoder_decoder_kwargs_for_generation
model_kwargs["encoder_outputs"]: ModelOutput = encoder(**encoder_kwargs)
File "C:\Python39\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Python39\lib\site-packages\transformers\models\t5\modeling_t5.py", line 988, in forward
inputs_embeds = self.embed_tokens(input_ids)
File "C:\Python39\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Python39\lib\site-packages\torch\nn\modules\sparse.py", line 162, in forward
return F.embedding(
File "C:\Python39\lib\site-packages\torch\nn\functional.py", line 2210, in embedding
return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
IndexError: index out of range in self

Sign up or log in to comment