Charbel Malo commited on
Commit
f98a083
·
verified ·
1 Parent(s): 6907611

Update face_swapper.py

Browse files
Files changed (1) hide show
  1. face_swapper.py +1 -1
face_swapper.py CHANGED
@@ -42,7 +42,7 @@ def norm_crop2(img, landmark, image_size=112, mode='arcface'):
42
 
43
 
44
  class Inswapper():
45
- def __init__(self, model_file=None, batch_size=32, providers=['CPUExecutionProvider']):
46
  self.model_file = model_file
47
  self.batch_size = batch_size
48
 
 
42
 
43
 
44
  class Inswapper():
45
+ def __init__(self, model_file=None, batch_size=32, providers=['CUDAExecutionProvider']):
46
  self.model_file = model_file
47
  self.batch_size = batch_size
48