sczhou commited on
Commit
7cc7518
·
1 Parent(s): 637d603

V0.1.0 release.

Browse files
Files changed (1) hide show
  1. inference_codeformer.py +1 -1
inference_codeformer.py CHANGED
@@ -26,7 +26,7 @@ if __name__ == '__main__':
26
  parser.add_argument('--has_aligned', action='store_true', help='Input are cropped and aligned faces')
27
  parser.add_argument('--only_center_face', action='store_true', help='Only restore the center face')
28
  parser.add_argument('--draw_box', action='store_true')
29
- parser.add_argument('--bg_upsampler', type=str, default='realesrgan', help='background upsampler. Default: realesrgan')
30
  parser.add_argument('--bg_tile', type=int, default=400, help='Tile size for background sampler. Default: 400')
31
 
32
  args = parser.parse_args()
 
26
  parser.add_argument('--has_aligned', action='store_true', help='Input are cropped and aligned faces')
27
  parser.add_argument('--only_center_face', action='store_true', help='Only restore the center face')
28
  parser.add_argument('--draw_box', action='store_true')
29
+ parser.add_argument('--bg_upsampler', type=str, default='None', help='background upsampler. Optional: realesrgan')
30
  parser.add_argument('--bg_tile', type=int, default=400, help='Tile size for background sampler. Default: 400')
31
 
32
  args = parser.parse_args()