Merge pull request #69 from MarcusLoppe/sageconv_dropout_fix

Removed sageconv dropout due to not existing anymore(?)

Author: lucidrains

meshgpt_pytorch/meshgpt_pytorch.py

@@ -433,7 +433,6 @@ def __init__(
         use_linear_attn = True,
         pad_id = -1,
         flash_attn = True,
-        sageconv_dropout = 0.,
         attn_dropout = 0.,
         ff_dropout = 0.,
         resnet_dropout = 0,
@@ -482,10 +481,7 @@ def __init__(
 
         self.project_in = nn.Linear(init_dim, dim_codebook)
 
-        # initial sage conv
-
-        sageconv_kwargs = {**sageconv_kwargs, 'sageconv_dropout' : sageconv_dropout}
-
+        # initial sage conv 
         init_encoder_dim, *encoder_dims_through_depth = encoder_dims_through_depth
         curr_dim = init_encoder_dim
 

meshgpt_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.1.0'
+__version__ = '1.1.1'