stable-diffusion-webui

1
from modules import shared
2

3

4
class FaceRestoration:
5
    def name(self):
6
        return "None"
7

8
    def restore(self, np_image):
9
        return np_image
10

11

12
def restore_faces(np_image):
13
    face_restorers = [x for x in shared.face_restorers if x.name() == shared.opts.face_restoration_model or shared.opts.face_restoration_model is None]
14
    if len(face_restorers) == 0:
15
        return np_image
16

17
    face_restorer = face_restorers[0]
18

19
    return face_restorer.restore(np_image)
20