update

app.py

@@ -316,7 +316,7 @@ with gr.Blocks() as demo:
                     inputs=[input_image],
                     label="Examples",
                     cache_examples=False,
-                    examples_per_page=15
+                    examples_per_page=12
                 )
 
         with gr.Column():

src/models/renderer/utils/renderer.py

@@ -68,7 +68,7 @@ def sample_from_planes(plane_axes, plane_features, coordinates, mode='bilinear',
 
     coordinates = (2/box_warp) * coordinates # add specific box bounds
 
-    print('plane_axes', plane_axes.dtype, 'plane_features', plane_features.dtype, 'coordinates', coordinates.dtype)
+    print('plane_axes', plane_axes.device, 'plane_features', plane_features.device, 'coordinates', coordinates.device)
 
     projected_coordinates = project_onto_planes(plane_axes, coordinates).unsqueeze(1)
     output_features = torch.nn.functional.grid_sample(