Paper99 commited on
Commit
0e6b5f9
1 Parent(s): a79c181

Update PhotoMaker v2 demo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ from aspect_ratio_template import aspect_ratios
20
 
21
  # global variable
22
  base_model_path = 'SG161222/RealVisXL_V4.0'
23
- face_detector = FaceAnalysis2(providers=['CUDAExecutionProvider'], allowed_modules=['detection', 'recognition'])
24
  face_detector.prepare(ctx_id=0, det_size=(640, 640))
25
 
26
  try:
 
20
 
21
  # global variable
22
  base_model_path = 'SG161222/RealVisXL_V4.0'
23
+ face_detector = FaceAnalysis2(providers=['CPUExecutionProvider', 'CUDAExecutionProvider'], allowed_modules=['detection', 'recognition'])
24
  face_detector.prepare(ctx_id=0, det_size=(640, 640))
25
 
26
  try: