Hugging Face's logo

Spaces:
mkings
/
simple_lama
Sleeping

App Files Community
Kingtous commited on
Commit
b3929b6
·
1 Parent(s): d04db90

feat: cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
  import cv2
8
  from simple_lama_inpainting import SimpleLama
9
 
10
- zero = torch.Tensor([0])
11
  print(zero.device) # <-- 'cpu' 🤔
12
  lama: SimpleLama = SimpleLama(device=zero.device)
13
 
 
7
  import cv2
8
  from simple_lama_inpainting import SimpleLama
9
 
10
+ zero = torch.Tensor([0]).cuda()
11
  print(zero.device) # <-- 'cpu' 🤔
12
  lama: SimpleLama = SimpleLama(device=zero.device)
13