FIX: "not enabled" logic change

This commit is contained in:
Gourieff 2023-08-19 01:03:05 +07:00
parent aea92f7e15
commit 8c4454d295

View File

@ -198,13 +198,15 @@ class FaceSwapScript(scripts.Script):
gender_source, gender_source,
gender_target, gender_target,
): ):
self.enable = enable
if self.enable:
global MODELS_PATH global MODELS_PATH
self.source = img self.source = img
self.face_restorer_name = face_restorer_name self.face_restorer_name = face_restorer_name
self.upscaler_scale = upscaler_scale self.upscaler_scale = upscaler_scale
self.upscaler_visibility = upscaler_visibility self.upscaler_visibility = upscaler_visibility
self.face_restorer_visibility = face_restorer_visibility self.face_restorer_visibility = face_restorer_visibility
self.enable = enable
self.restore_first = restore_first self.restore_first = restore_first
self.upscaler_name = upscaler_name self.upscaler_name = upscaler_name
self.swap_in_generated = swap_in_generated self.swap_in_generated = swap_in_generated
@ -222,7 +224,7 @@ class FaceSwapScript(scripts.Script):
self.source_faces_index = [0] self.source_faces_index = [0]
if len(self.faces_index) == 0: if len(self.faces_index) == 0:
self.faces_index = [0] self.faces_index = [0]
if self.enable:
if self.source is not None: if self.source is not None:
apply_logging_patch(console_logging_level) apply_logging_patch(console_logging_level)
if isinstance(p, StableDiffusionProcessingImg2Img) and swap_in_source: if isinstance(p, StableDiffusionProcessingImg2Img) and swap_in_source: