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 |
|