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'
|