Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,10 +6,10 @@ from accelerate import init_empty_weights, load_checkpoint_and_dispatch
|
|
| 6 |
# Detección y configuración del dispositivo para compatibilidad con GPU o CPU
|
| 7 |
if torch.cuda.is_available():
|
| 8 |
device = "cuda" # Para GPUs NVIDIA
|
| 9 |
-
elif torch.backends.mps.
|
| 10 |
device = "mps" # Para GPUs Apple Silicon (M1/M2) y otras GPUs con soporte Metal
|
| 11 |
-
elif torch.backends.rocm.is_available():
|
| 12 |
-
device = "rocm" # Para GPUs AMD con ROCm
|
| 13 |
else:
|
| 14 |
device = "cpu" # En caso de no tener GPU disponible
|
| 15 |
|
|
|
|
| 6 |
# Detección y configuración del dispositivo para compatibilidad con GPU o CPU
|
| 7 |
if torch.cuda.is_available():
|
| 8 |
device = "cuda" # Para GPUs NVIDIA
|
| 9 |
+
elif hasattr(torch.backends, "mps") and torch.backends.mps.is_built():
|
| 10 |
device = "mps" # Para GPUs Apple Silicon (M1/M2) y otras GPUs con soporte Metal
|
| 11 |
+
elif hasattr(torch.backends, "rocm") and torch.backends.rocm.is_available():
|
| 12 |
+
device = "rocm" # Para GPUs AMD con ROCm, si está disponible
|
| 13 |
else:
|
| 14 |
device = "cpu" # En caso de no tener GPU disponible
|
| 15 |
|