Tolga commited on
Commit
b79be99
1 Parent(s): 7354fac
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ import time
17
  warnings.filterwarnings("ignore")
18
 
19
  device = torch.device('cuda:0')
20
- device = torch.device('cpu')
21
 
22
 
23
  dataset_name = 'vox' # ['vox', 'taichi', 'ted', 'mgif']
 
17
  warnings.filterwarnings("ignore")
18
 
19
  device = torch.device('cuda:0')
20
+ #device = torch.device('cpu')
21
 
22
 
23
  dataset_name = 'vox' # ['vox', 'taichi', 'ted', 'mgif']