update

nsr/script_util.py

@@ -19,7 +19,7 @@ from ldm.modules.diffusionmodules.model import Encoder, Decoder, MVEncoder, MVEn
 from ldm.modules.diffusionmodules.mv_unet import MVUNet, LGM_MVEncoder
 from torch.profiler import profile, record_function, ProfilerActivity
 
-from nsr.gs import GaussianRenderer
+# from nsr.gs import GaussianRenderer
 from nsr.gs_surfel import GaussianRenderer2DGS
 # from nsr.srt.encoder import ImprovedSRTEncoderVAE, ImprovedSRTEncoderVAE_L5_vitl, ImprovedSRTEncoderVAE_mlp_ratio4, ImprovedSRTEncoderVAE_L6, ImprovedSRTEncoderVAE_mlp_ratio4_f8, ImprovedSRTEncoderVAE_mlp_ratio4_heavyPatchify, ImprovedSRTEncoderVAE_mlp_ratio4_f8_L6, ImprovedSRTEncoderVAE_mlp_ratio4_L6, HybridEncoder, ImprovedSRTEncoderVAE_mlp_ratio4_decomposed, HybridEncoderPCDStructuredLatent
 from nsr.srt.encoder import *
@@ -1482,9 +1482,9 @@ def create_3DAE_model(
         elif surfel_rendering:
             triplane_decoder = GaussianRenderer2DGS(
                 image_size, out_chans, rendering_kwargs=rendering_kwargs)
-        else:
-            triplane_decoder = GaussianRenderer(
-                image_size, out_chans, rendering_kwargs=rendering_kwargs)
+        # else:
+        #     triplane_decoder = GaussianRenderer(
+        #         image_size, out_chans, rendering_kwargs=rendering_kwargs)
 
         if load_pretrain_encoder:
 

vit/vit_triplane.py

@@ -58,7 +58,7 @@ from timm.models.vision_transformer import PatchEmbed
 
 from utils.general_utils import matrix_to_quaternion, quaternion_raw_multiply, build_rotation
 
-from nsr.gs import GaussianRenderer
+# from nsr.gs import GaussianRenderer
 
 from utils.dust3r.heads import create_dpt_head