nguyenp99 commited on
Commit
c5d06ca
1 Parent(s): a8acdb5

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -7,7 +7,7 @@ class Config(f.Pipe):
7
 
8
  def __init__(self) -> None:
9
  super().__init__(
10
- DEVICE = 'cuda:0',
11
 
12
  STANDARD_WIDTH = 1024,
13
  STANDARD_INTERPOLATION = INTER_CUBIC,
 
7
 
8
  def __init__(self) -> None:
9
  super().__init__(
10
+ DEVICE = 'cuda:1',
11
 
12
  STANDARD_WIDTH = 1024,
13
  STANDARD_INTERPOLATION = INTER_CUBIC,