NouFuS commited on
Commit
0c6674b
1 Parent(s): fc52adb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -4,7 +4,6 @@ import torch
4
  import numpy as np
5
 
6
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
7
-
8
  #device = "cpu"
9
  torch_dtype = torch.float16 if device != "cpu" else torch.float32
10
 
 
4
  import numpy as np
5
 
6
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
7
  #device = "cpu"
8
  torch_dtype = torch.float16 if device != "cpu" else torch.float32
9