Spaces:
Running
Running
Ffftdtd5dtft
commited on
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 |
|