roman-bachmann commited on
Commit
f9de43d
·
verified ·
1 Parent(s): f8de4d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -28,6 +28,8 @@ if torch.cuda.is_available():
28
  print(f'Device: {device}, GPU type: {gpu_type}')
29
  print('BF16 enabled:', enable_bf16)
30
  else:
 
 
31
  device, power_device, enable_bf16 = "cpu", "CPU", False
32
  print('Running on CPU')
33
 
 
28
  print(f'Device: {device}, GPU type: {gpu_type}')
29
  print('BF16 enabled:', enable_bf16)
30
  else:
31
+ import torch._dynamo
32
+ torch._dynamo.config.suppress_errors = True
33
  device, power_device, enable_bf16 = "cpu", "CPU", False
34
  print('Running on CPU')
35