fix alignment bug
This commit is contained in:
parent
a35a2a15f3
commit
f5b3356d62
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue