Commit 758a7fdc authored by dmMaze's avatar dmMaze
Browse files

fix inpaint

parent 95ec4953

modules/inpaint/base.py

+4 −2
Original line number Diff line number Diff line
@@ -270,6 +270,7 @@ class AOTInpainter(InpainterBase): 


        if param_key == 'device':

            param_device = self.params['device']['select']

            if self.model is not None:

                self.model.to(param_device)

            self.device = param_device
@@ -386,6 +387,7 @@ class LamaInpainterMPE(InpainterBase): 


        if param_key == 'device':

            param_device = self.params['device']['select']

            if self.model is not None:

                self.model.to(param_device)

            self.device = param_device