Upload app.py
Browse files
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
|