This commit is contained in:
Woosuk Kwon 2023-02-23 20:19:24 +00:00
parent 1f6c7ef437
commit 86f9eb6d39

View File

@ -179,7 +179,6 @@ class OPTDecoder(OPTPreTrainedModel):
) -> torch.Tensor:
inputs_embeds = self.embed_tokens(input_ids)
pos_embeds = self.embed_positions(positions)
pos_embeds = None
if self.project_in is not None:
inputs_embeds = self.project_in(inputs_embeds)
hidden_states = inputs_embeds + pos_embeds