Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -30,11 +30,10 @@ seed = random.randint(1, 100000)
|
|
30 |
torch.manual_seed(seed)
|
31 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
32 |
dtype = 'bfloat16' if torch.cuda.is_available() and torch.cuda.is_bf16_supported() else 'float16'
|
33 |
-
compile =
|
34 |
exec(open('configurator.py').read())
|
35 |
|
36 |
-
|
37 |
-
torch.cuda.manual_seed(seed)
|
38 |
torch.backends.cuda.matmul.allow_tf32 = True
|
39 |
torch.backends.cudnn.allow_tf32 = True
|
40 |
device_type = 'cpu' if 'cuda' in device else 'cpu'
|
|
|
30 |
torch.manual_seed(seed)
|
31 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
32 |
dtype = 'bfloat16' if torch.cuda.is_available() and torch.cuda.is_bf16_supported() else 'float16'
|
33 |
+
compile = True
|
34 |
exec(open('configurator.py').read())
|
35 |
|
36 |
+
|
|
|
37 |
torch.backends.cuda.matmul.allow_tf32 = True
|
38 |
torch.backends.cudnn.allow_tf32 = True
|
39 |
device_type = 'cpu' if 'cuda' in device else 'cpu'
|