allenai / allenai/longformer

index out of range in self!

Abierto
#163 6 comentarios 0 reacciones 0 asignados Ver en GitHub
Lenguaje dominante
Python
Estrellas
2.2k
Forks
286
Métricas de merge de PR
Sin PR fusionados en 30 d

Descripción

I want to use the pre-trained long-former model to get embeddings for long documents (maximum 5000 words). I am trying to execute the demo on relatively long documents to test the library. The code works if the input text is small. But it gives the following error when executing it for long documents. any help?

**P.S I am running it on a regular CPU**
```
config = LongformerConfig.from_pretrained('longformer-encdec-base-16384/')
config.attention_mode = 'sliding_chunks'
model = LongformerModel.from_pretrained("longformer-encdec-base-16384/")
tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
tokenizer.model_max_length = model.config.max_position_embeddings
SAMPLE_TEXT = ' '.join(['Hello to this world'] * 2000) # long input document

input_ids = torch.tensor(tokenizer.encode(SAMPLE_TEXT)).unsqueeze(0)

attention_mask = torch.ones(input_ids.shape, dtype=torch.long,
device=input_ids.device) # initialize to local attention

input_ids, attention_mask = pad_to_window_size(
input_ids, attention_mask, config.attention_window[0], tokenizer.pad_token_id)
x = model(input_ids, attention_mask=attention_mask)[0]
print(x)
```

> x = modelLong(input_ids, attention_mask=attention_mask)
> File "/Users/omarsayed/PycharmProjects/testLongFormer/venv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
> result = self.forward(*input, **kwargs)
> File "/Users/omarsayed/PycharmProjects/testLongFormer/venv/lib/python3.8/site-packages/transformers/modeling_longformer.py", line 1070, in forward
> embedding_output = self.embeddings(
> File "/Users/omarsayed/PycharmProjects/testLongFormer/venv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
> result = self.forward(*input, **kwargs)
> File "/Users/omarsayed/PycharmProjects/testLongFormer/venv/lib/python3.8/site-packages/transformers/modeling_roberta.py", line 81, in forward
> return super().forward(
> File "/Users/omarsayed/PycharmProjects/testLongFormer/venv/lib/python3.8/site-packages/transformers/modeling_bert.py", line 208, in forward
> position_embeddings = self.position_embeddings(position_ids)
> File "/Users/omarsayed/PycharmProjects/testLongFormer/venv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
> result = self.forward(*input, **kwargs)
> File "/Users/omarsayed/PycharmProjects/testLongFormer/venv/lib/python3.8/site-packages/torch/nn/modules/sparse.py", line 124, in forward
> return F.embedding(
> File "/Users/omarsayed/PycharmProjects/testLongFormer/venv/lib/python3.8/site-packages/torch/nn/functional.py", line 1814, in embedding
> return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
> IndexError: index out of range in self

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.