fix alignment bug

This commit is contained in:
Yiwen Chen 2024-08-05 10:20:46 +08:00
parent a35a2a15f3
commit f5b3356d62
1 changed files with 1 additions and 0 deletions

View File

@ -327,6 +327,7 @@ class ShapeOPTDecoder(OPTDecoder):
dtype=inputs_embeds.dtype).long()
inputs_embeds = inputs_embeds + self.cond_embed(cond_embed_query)
elif inputs_embeds is not None:
# assert self.cond and not self.training