guigonzalez commited on
Commit
693c0f5
·
verified ·
1 Parent(s): bd3a53e

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -81,9 +81,11 @@ from triposg.pipelines.pipeline_triposg import TripoSGPipeline
81
  from briarmbg import BriaRMBG
82
 
83
  # Determinar dispositivo
 
 
84
  if torch.cuda.is_available():
85
  DEVICE = "cuda"
86
- DTYPE = torch.float16
87
  elif torch.backends.mps.is_available():
88
  DEVICE = "mps"
89
  DTYPE = torch.float32
 
81
  from briarmbg import BriaRMBG
82
 
83
  # Determinar dispositivo
84
+ # IMPORTANTE: Usar float32 mesmo em CUDA para evitar erro de dtype mismatch
85
+ # na função hierarchical_extract_geometry (mat1 Float vs mat2 Half)
86
  if torch.cuda.is_available():
87
  DEVICE = "cuda"
88
+ DTYPE = torch.float32 # float16 causa erro de dtype mismatch
89
  elif torch.backends.mps.is_available():
90
  DEVICE = "mps"
91
  DTYPE = torch.float32