update

.github/workflows/python-publish.yml

@@ -0,0 +1,38 @@
+
+
+# This workflow will upload a Python Package using Twine when a release is created
+# For more information see: https://help.github.com/en/actions/language-and-framework-guides/using-python-with-github-actions#publishing-to-package-registries
+
+# This workflow uses actions that are not certified by GitHub.
+# They are provided by a third-party and are governed by
+# separate terms of service, privacy policy, and support
+# documentation.
+
+name: Upload Python Package
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  deploy:
+
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python
+      uses: actions/setup-python@v2
+      with:
+        python-version: '3.x'
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install build
+    - name: Build package
+      run: python -m build
+    - name: Publish package
+      uses: pypa/[email protected]
+      with:
+        user: __token__
+        password: ${{ secrets.PYPI_API_TOKEN }}

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.4'

setup.py

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