Skip to content

Commit 047bfbf

Browse files
committed
fix an issue
with text embedding padding, thanks to @MarcusLoppe
1 parent 837449f commit 047bfbf

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

meshgpt_pytorch/meshgpt_pytorch.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1122,7 +1122,8 @@ def __init__(
11221122
if condition_on_text:
11231123
self.conditioner = TextEmbeddingReturner(
11241124
model_types = text_condition_model_types,
1125-
cond_drop_prob = text_condition_cond_drop_prob
1125+
cond_drop_prob = text_condition_cond_drop_prob,
1126+
text_embed_pad_value = -1.
11261127
)
11271128

11281129
dim_text = self.conditioner.dim_latent

meshgpt_pytorch/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = '1.2.15'
1+
__version__ = '1.2.16'

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
install_requires=[
2323
'accelerate>=0.25.0',
2424
'beartype',
25-
'classifier-free-guidance-pytorch>=0.5.1',
25+
'classifier-free-guidance-pytorch>=0.6.0',
2626
'einops>=0.7.0',
2727
'einx[torch]>=0.1.3',
2828
'ema-pytorch',

0 commit comments

Comments
 (0)