Update config.py
Browse files
config.py
CHANGED
@@ -7,7 +7,7 @@ class Config(f.Pipe):
|
|
7 |
|
8 |
def __init__(self) -> None:
|
9 |
super().__init__(
|
10 |
-
DEVICE = 'cuda:
|
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,
|