GeorgiosIoannouCoder commited on
Commit
027302d
1 Parent(s): 41f4b73

Update inference/real_esrgan.py

Browse files
Files changed (1) hide show
  1. 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("cuda" if torch.cuda.is_available() else "cpu")
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
  )