honeymoose
/
MeshAnything
mirror of https://github.com/buaacyw/MeshAnything.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
MeshAnything/models/shape_opt.py
View File

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