Update app.py

app.py

@@ -347,7 +347,7 @@ def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[s
         # GLB 변환
         with torch.inference_mode():
             try:
-                # 모든 텐서를 CUDA로 이동하고 gradient 활성화
+                # 모든 텐서를 CUDA로 이동 (gradient 불필요)
                 device = torch.device('cuda:0')
                 
                 # Gaussian 텐서들을 변환
@@ -355,24 +355,23 @@ def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[s
                     if hasattr(gs, attr_name):
                         tensor = getattr(gs, attr_name)
                         if torch.is_tensor(tensor):
-                            # gradient 계산이 필요한 텐서로 변환
-                            new_tensor = tensor.detach().clone().float().to(device).requires_grad_(True)
+                            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().to(device).requires_grad_(True)
+                    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().to(device)
                 
-                # 추가 속성 확인 및 변환
+                # 추가 속성 확인 및 변환 (gradient 불필요)
                 for attr_name in dir(mesh):
                     if attr_name.startswith('_'):
                         continue
                     attr = getattr(mesh, attr_name)
                     if torch.is_tensor(attr):
                         if attr.dtype in [torch.float32, torch.float64]:
-                            setattr(mesh, attr_name, attr.to(device).requires_grad_(True))
+                            setattr(mesh, attr_name, attr.to(device))
                         else:
                             setattr(mesh, attr_name, attr.to(device))
                 
@@ -425,6 +424,7 @@ def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[s
         clear_gpu_memory()
 
 
+
 def activate_button() -> gr.Button:
     return gr.Button(interactive=True)