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

Update face_enhancer.py

Browse files
Files changed (1) hide show
  1. face_enhancer.py +1 -1
face_enhancer.py CHANGED
@@ -40,7 +40,7 @@ def get_available_enhancer_names():
40
  return available
41
 
42
 
43
- def load_face_enhancer_model(name='GFPGAN', device="cpu"):
44
  assert name in get_available_enhancer_names() + cv2_interpolations, f"Face enhancer {name} unavailable."
45
  if name in supported_enhancers.keys():
46
  model_path, model_runner = supported_enhancers.get(name)
 
40
  return available
41
 
42
 
43
+ def load_face_enhancer_model(name='GFPGAN', device="cuda"):
44
  assert name in get_available_enhancer_names() + cv2_interpolations, f"Face enhancer {name} unavailable."
45
  if name in supported_enhancers.keys():
46
  model_path, model_runner = supported_enhancers.get(name)