Update app.py

app.py

@@ -347,22 +347,36 @@ def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[s
         # GLB 변환
         with torch.inference_mode():
             try:
-                # Gaussian 텐서들을 float32로 변환
+                # 모든 텐서를 CUDA로 이동
+                device = torch.device('cuda:0')
+                
+                # Gaussian 텐서들을 변환
                 for attr_name in ['_xyz', '_features_dc', '_scaling', '_rotation', '_opacity']:
                     if hasattr(gs, attr_name):
                         tensor = getattr(gs, attr_name)
                         if torch.is_tensor(tensor):
-                            # gradient 추적 제거 및 float32로 변환
-                            new_tensor = tensor.detach().clone().float().cpu()
+                            new_tensor = tensor.detach().clone().float().to(device)
                             setattr(gs, attr_name, new_tensor)
                 
                 # Mesh 텐서들을 변환
                 if hasattr(mesh, 'vertices') and torch.is_tensor(mesh.vertices):
-                    mesh.vertices = mesh.vertices.detach().clone().float().cpu()
+                    mesh.vertices = mesh.vertices.detach().clone().float().to(device)
                 if hasattr(mesh, 'faces') and torch.is_tensor(mesh.faces):
-                    mesh.faces = mesh.faces.detach().clone().long().cpu()
+                    mesh.faces = mesh.faces.detach().clone().long().to(device)
+                
+                # 추가 속성 확인 및 변환
+                for attr_name in dir(mesh):
+                    if attr_name.startswith('_'):
+                        continue
+                    attr = getattr(mesh, attr_name)
+                    if torch.is_tensor(attr):
+                        setattr(mesh, attr_name, attr.to(device))
+                
+                print("Device check before GLB conversion:")
+                print(f"Gaussian xyz device: {gs._xyz.device}")
+                print(f"Mesh vertices device: {mesh.vertices.device}")
                 
-                # 기본 설정으로 GLB 변환
+                # GLB 변환
                 glb = postprocessing_utils.to_glb(
                     gs, 
                     mesh, 
@@ -373,6 +387,11 @@ def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[s
                 
             except Exception as e:
                 print(f"Error during GLB conversion: {str(e)}")
+                # 디바이스 정보 출력
+                if hasattr(gs, '_xyz'):
+                    print(f"Gaussian xyz device: {gs._xyz.device}")
+                if hasattr(mesh, 'vertices'):
+                    print(f"Mesh vertices device: {mesh.vertices.device}")
                 return None, None
             
         if glb is None: