YulianSa commited on
Commit
139b771
·
1 Parent(s): 12d3833
pre-requirements.txt CHANGED
@@ -23,7 +23,6 @@ local_attention==1.9.15
23
  taylor_series_linear_attention
24
  transformers
25
  vector_quantize_pytorch
26
- open3d
27
  trimesh
28
  numpy==1.26.4
29
  pytorch_lightning
 
23
  taylor_series_linear_attention
24
  transformers
25
  vector_quantize_pytorch
 
26
  trimesh
27
  numpy==1.26.4
28
  pytorch_lightning
primitive_anything/primitive_dataset.py CHANGED
@@ -5,7 +5,6 @@ import os
5
  import numpy as np
6
  from scipy.linalg import polar
7
  from scipy.spatial.transform import Rotation
8
- import open3d as o3d
9
  import torch
10
  from torch.utils.data import Dataset
11
 
 
5
  import numpy as np
6
  from scipy.linalg import polar
7
  from scipy.spatial.transform import Rotation
 
8
  import torch
9
  from torch.utils.data import Dataset
10
 
primitive_anything/primitive_transformer.py CHANGED
@@ -11,7 +11,6 @@ from einops import rearrange, repeat, reduce, pack
11
  from gateloop_transformer import SimpleGateLoopLayer
12
  from huggingface_hub import PyTorchModelHubMixin
13
  import numpy as np
14
- import open3d as o3d
15
  from tqdm import tqdm
16
  import torch
17
  from torch import nn, Tensor
 
11
  from gateloop_transformer import SimpleGateLoopLayer
12
  from huggingface_hub import PyTorchModelHubMixin
13
  import numpy as np
 
14
  from tqdm import tqdm
15
  import torch
16
  from torch import nn, Tensor