allow for setting text model adapter kwargs at CFG lib

Author: lucidrains

meshgpt_pytorch/meshgpt_pytorch.py

@@ -1111,6 +1111,7 @@ def __init__(
         condition_on_text = False,
         text_cond_with_film = False,
         text_condition_model_types = ('t5',),
+        text_condition_model_kwargs = (dict(),),
         text_condition_cond_drop_prob = 0.25,
         quads = False,
     ):
@@ -1162,6 +1163,7 @@ def __init__(
         if condition_on_text:
             self.conditioner = TextEmbeddingReturner(
                 model_types = text_condition_model_types,
+                model_kwargs = text_condition_model_kwargs,
                 cond_drop_prob = text_condition_cond_drop_prob,
                 text_embed_pad_value = -1.
             )

meshgpt_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.5.2'
+__version__ = '1.5.3'

setup.py

@@ -23,7 +23,7 @@
     'accelerate>=0.25.0',
     'beartype',
     "huggingface_hub>=0.21.4",
-    'classifier-free-guidance-pytorch>=0.6.7',
+    'classifier-free-guidance-pytorch>=0.6.8',
     'einops>=0.8.0',
     'einx[torch]>=0.3.0',
     'ema-pytorch>=0.5.1',