sczhou commited on
Commit
9715dfb
·
1 Parent(s): 4c93a0a

uncomment of face_upsampler.

Browse files
Files changed (1) hide show
  1. inference_codeformer.py +4 -5
inference_codeformer.py CHANGED
@@ -123,11 +123,10 @@ if __name__ == '__main__':
123
 
124
  # ------------------ set up face upsampler ------------------
125
  if args.face_upsample:
126
- face_upsampler = None
127
- # if bg_upsampler is not None:
128
- # face_upsampler = bg_upsampler
129
- # else:
130
- # face_upsampler = set_realesrgan()
131
  else:
132
  face_upsampler = None
133
 
 
123
 
124
  # ------------------ set up face upsampler ------------------
125
  if args.face_upsample:
126
+ if bg_upsampler is not None:
127
+ face_upsampler = bg_upsampler
128
+ else:
129
+ face_upsampler = set_realesrgan()
 
130
  else:
131
  face_upsampler = None
132