Save lfs.
Browse files- .gitignore +1 -4
- transformer_directory/model.safetensors → output.bin +2 -2
- predict.py +2 -8
.gitignore
CHANGED
@@ -1,8 +1,5 @@
|
|
1 |
-
|
2 |
render.obj
|
3 |
-
|
4 |
.cog/
|
5 |
-
|
6 |
__pycache__/
|
7 |
-
|
8 |
output.obj
|
|
|
|
|
|
1 |
render.obj
|
|
|
2 |
.cog/
|
|
|
3 |
__pycache__/
|
|
|
4 |
output.obj
|
5 |
+
output.bin
|
transformer_directory/model.safetensors → output.bin
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ba39696e8eac4236b332aba9ecc4232831ef5ef12ad44b83d0e894766b9dd555
|
3 |
+
size 25986
|
predict.py
CHANGED
@@ -2,11 +2,12 @@ from cog import BasePredictor, Input, Path
|
|
2 |
import tempfile
|
3 |
import time
|
4 |
from meshgpt_pytorch import MeshTransformer, mesh_render
|
|
|
5 |
|
6 |
class Predictor(BasePredictor):
|
7 |
def setup(self):
|
8 |
"""Load the model into memory to make running multiple predictions efficient"""
|
9 |
-
self.transformer = MeshTransformer.from_pretrained("
|
10 |
|
11 |
def predict(
|
12 |
self,
|
@@ -51,10 +52,3 @@ class Predictor(BasePredictor):
|
|
51 |
file_path = Path(tempfile.mkdtemp()) / file_name
|
52 |
mesh_render.save_rendering(str(file_path), output)
|
53 |
return file_path
|
54 |
-
|
55 |
-
|
56 |
-
if __name__ == "__main__":
|
57 |
-
model_name = "MarcusLoren/MeshGPT-preview"
|
58 |
-
|
59 |
-
model = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
|
60 |
-
model.save_pretrained("./transformer_directory")
|
|
|
2 |
import tempfile
|
3 |
import time
|
4 |
from meshgpt_pytorch import MeshTransformer, mesh_render
|
5 |
+
from os.path import dirname
|
6 |
|
7 |
class Predictor(BasePredictor):
|
8 |
def setup(self):
|
9 |
"""Load the model into memory to make running multiple predictions efficient"""
|
10 |
+
self.transformer = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
|
11 |
|
12 |
def predict(
|
13 |
self,
|
|
|
52 |
file_path = Path(tempfile.mkdtemp()) / file_name
|
53 |
mesh_render.save_rendering(str(file_path), output)
|
54 |
return file_path
|
|
|
|
|
|
|
|
|
|
|
|
|
|