fix alignment bug

2024-08-05 10:20:46 +08:00 · 2024-08-05 10:20:46 +08:00 · f5b3356d62
parent a35a2a15f3
commit f5b3356d62
1 changed files with 1 additions and 0 deletions
--- a/MeshAnything/models/shape_opt.py
+++ b/MeshAnything/models/shape_opt.py
@ -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