Spaces:

EPFL-VILAB
/

FlexTok

Running on Zero

roman-bachmann commited on Apr 1

Commit

7d2453d

verified ·

1 Parent(s): f9de43d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,8 +28,7 @@ if torch.cuda.is_available():
     print(f'Device: {device}, GPU type: {gpu_type}')
     print('BF16 enabled:', enable_bf16)
 else:
-    import torch._dynamo
-    torch._dynamo.config.suppress_errors = True
     device, power_device, enable_bf16 = "cpu", "CPU", False
     print('Running on CPU')

     print(f'Device: {device}, GPU type: {gpu_type}')
     print('BF16 enabled:', enable_bf16)
 else:
+    # Currently not supported. Please run on GPUs.
     device, power_device, enable_bf16 = "cpu", "CPU", False
     print('Running on CPU')