Spaces:
Runtime error
Runtime error
uncomment of face_upsampler.
Browse files- 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 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 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 |
|