add dubeg info

app.py

@@ -35,7 +35,8 @@ GRADIO_OBJ_SHADING_PATH = 'gradio_output_shading.obj'
 #opt = tyro.cli(AllConfigs)
 #torch._dynamo.config.verbose=True
 
-ckpt_path = hf_hub_download(repo_id="rgxie/LDM", filename="LDM6v01.ckpt")
+#ckpt_path = hf_hub_download(repo_id="rgxie/LDM", filename="LDM6v01.ckpt")
+ckpt_path='/ssd3/xrg/tensor23d/pretrained/last_6view_0610_14.ckpt'
 
 opt = Options(
     input_size=512, 

core/models.py

@@ -122,8 +122,9 @@ class LTRFM_NeRF(nn.Module):
         xyz_samples=xyz_samples.permute(0,2,3,4,1)
         xyz_samples=xyz_samples.view(1,-1,1,3)
         
-        print("plane"+ planes.device)
-        print("xyz_samples"+ xyz_samples.device)
+        print("app and xyz")
+        print(planes['app_planes'].device)
+        print(xyz_samples.device)
 
         grid_out = self.tensorRF.predict_sdf(planes,xyz_samples)
         grid_out['sigma']=grid_out['sigma'].view(grid_size,grid_size,grid_size).float()