GeorgiosIoannouCoder
commited on
Commit
•
027302d
1
Parent(s):
41f4b73
Update inference/real_esrgan.py
Browse files- inference/real_esrgan.py +1 -1
inference/real_esrgan.py
CHANGED
@@ -72,7 +72,7 @@ class RealEsrGan:
|
|
72 |
)
|
73 |
else:
|
74 |
self.device = (
|
75 |
-
torch.device("
|
76 |
if self.device is None
|
77 |
else self.device
|
78 |
)
|
|
|
72 |
)
|
73 |
else:
|
74 |
self.device = (
|
75 |
+
torch.device("cpu" if torch.cuda.is_available() else "cpu")
|
76 |
if self.device is None
|
77 |
else self.device
|
78 |
)
|