patchbanks commited on
Commit
a275e2f
·
verified ·
1 Parent(s): c513211

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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 = False
34
  exec(open('configurator.py').read())
35
 
36
- torch.manual_seed(seed)
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'