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: ) -> torch.Tensor:
inputs_embeds = self.embed_tokens(input_ids) inputs_embeds = self.embed_tokens(input_ids)
pos_embeds = self.embed_positions(positions) pos_embeds = self.embed_positions(positions)
pos_embeds = None
if self.project_in is not None: if self.project_in is not None:
inputs_embeds = self.project_in(inputs_embeds) inputs_embeds = self.project_in(inputs_embeds)
hidden_states = inputs_embeds + pos_embeds hidden_states = inputs_embeds + pos_embeds