init

app.py

@@ -6,7 +6,6 @@ from gradio_model4dgs import Model4DGS
 import numpy
 import hashlib
 
-os.system('pip install -e ./simple-knn')
 os.system('pip install -e ./diff-gaussian-rasterization')
 os.system('pip install xformers==0.0.23 --no-deps --index-url https://download.pytorch.org/whl/cu118')
 

gaussian_model_4d.py

@@ -18,7 +18,6 @@ from utils.system_utils import mkdir_p
 from plyfile import PlyData, PlyElement
 from random import randint
 from utils.sh_utils import RGB2SH
-from simple_knn._C import distCUDA2
 from utils.graphics_utils import BasicPointCloud
 from utils.general_utils import strip_symmetric, build_scaling_rotation
 from scene.deformation import deform_network
@@ -289,6 +288,7 @@ class GaussianModel:
             self.active_sh_degree += 1
 
     def create_from_pcd(self, pcd : BasicPointCloud, spatial_lr_scale : float, time_line: int):
+        from simple_knn._C import distCUDA2
         self.spatial_lr_scale = spatial_lr_scale
         fused_point_cloud = torch.tensor(np.asarray(pcd.points)).float().cuda()
         fused_color = RGB2SH(torch.tensor(np.asarray(pcd.colors)).float().cuda())