Ffftdtd5dtft commited on
Commit
9934eed
·
verified ·
1 Parent(s): 9675628

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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.is_available():
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