Load the models.

.gitignore

@@ -4,3 +4,5 @@ render.obj
 .cog/
 
 __pycache__/
+
+output.obj

predict.py

@@ -3,13 +3,10 @@ import tempfile
 import time
 from meshgpt_pytorch import MeshTransformer, mesh_render
 
-
 class Predictor(BasePredictor):
     def setup(self):
         """Load the model into memory to make running multiple predictions efficient"""
-        self.transformer = MeshTransformer.from_pretrained(
-            "MarcusLoren/MeshGPT-preview"
-        )
+        self.transformer = MeshTransformer.from_pretrained("./transformer_directory")
 
     def predict(
         self,
@@ -54,3 +51,10 @@ class Predictor(BasePredictor):
         file_path = Path(tempfile.mkdtemp()) / file_name
         mesh_render.save_rendering(str(file_path), output)
         return file_path
+
+
+if __name__ == "__main__":
+    model_name = "MarcusLoren/MeshGPT-preview"
+
+    model = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
+    model.save_pretrained("./transformer_directory")

transformer_directory/README.md

@@ -0,0 +1,9 @@
+---
+tags:
+- pytorch_model_hub_mixin
+- model_hub_mixin
+---
+
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Library: [More Information Needed]
+- Docs: [More Information Needed]

transformer_directory/config.json

@@ -0,0 +1,28 @@
+{
+  "attn_depth": 12,
+  "attn_dim_head": 64,
+  "attn_heads": 12,
+  "attn_kwargs": {
+    "attn_num_mem_kv": 4,
+    "ff_glu": true
+  },
+  "coarse_pre_gateloop_depth": 2,
+  "condition_on_text": true,
+  "cross_attn_num_mem_kv": 4,
+  "dim": 768,
+  "dropout": 0.0,
+  "fine_attn_depth": 2,
+  "fine_attn_dim_head": 32,
+  "fine_attn_heads": 8,
+  "fine_cross_attend_text": true,
+  "fine_pre_gateloop_depth": 2,
+  "flash_attn": true,
+  "gateloop_use_heinsen": false,
+  "max_seq_len": 1500,
+  "num_sos_tokens": 1,
+  "pad_id": -1,
+  "quads": false,
+  "text_cond_with_film": false,
+  "text_condition_cond_drop_prob": 0.0,
+  "text_condition_model_types": "bge"
+}

transformer_directory/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2782fe5aab4625a1b595b311ceac27cd6198370f935998a8707dd81d6706fc12
+size 926663752