Skip to content

Commit

Permalink
fix default settings by not marking non managed fields as do_not_save
Browse files Browse the repository at this point in the history
  • Loading branch information
glucauze committed Aug 24, 2023
1 parent d0c56ae commit 2e718ee
Show file tree
Hide file tree
Showing 5 changed files with 10 additions and 14 deletions.
5 changes: 0 additions & 5 deletions preload.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,3 @@ def preload(parser: ArgumentParser) -> None:
choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
help="Set the log level (DEBUG, INFO, WARNING, ERROR, CRITICAL)",
)
parser.add_argument(
"--faceswaplab_gpu",
action="store_true",
help="Enable GPU if set, disable if not set",
)
3 changes: 0 additions & 3 deletions scripts/faceswaplab.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,9 +111,6 @@ def ui(self, is_img2img: bool) -> List[gr.components.Component]:
# If the order is modified, the before_process should be changed accordingly.

components = components + post_processing
# Ask sd to not store in ui-config.json
for component in components:
setattr(component, "do_not_save_to_config", True)

return components

Expand Down
2 changes: 1 addition & 1 deletion scripts/faceswaplab_globals.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
)

# Defining the version flag for the application
VERSION_FLAG: str = "v1.2.3"
VERSION_FLAG: str = "v1.2.4"
# Defining the path for 'sd-webui-faceswaplab' inside the 'extensions' directory
EXTENSION_PATH = os.path.join("extensions", "sd-webui-faceswaplab")

Expand Down
3 changes: 3 additions & 0 deletions scripts/faceswaplab_ui/faceswaplab_inpainting_ui.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,4 +65,7 @@ def face_inpainting_ui(
inpaiting_model,
]

for component in gradio_components:
setattr(component, "do_not_save_to_config", True)

return gradio_components
11 changes: 6 additions & 5 deletions scripts/faceswaplab_ui/faceswaplab_unit_ui.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ def faceswap_unit_advanced_options(
elem_id=f"{id_prefix}_face{unit_num}_erosion_factor",
)

return [
components = [
face_restorer_name,
face_restorer_visibility,
codeformer_weight,
Expand All @@ -91,6 +91,11 @@ def faceswap_unit_advanced_options(
erosion_factor,
]

for component in components:
setattr(component, "do_not_save_to_config", True)

return components


def faceswap_unit_ui(
is_img2img: bool, unit_num: int = 1, id_prefix: str = "faceswaplab"
Expand Down Expand Up @@ -281,10 +286,6 @@ def refresh_fn(selected: str):
+ post_inpainting
)

# Ask sd to not store in ui-config.json
for component in gradio_components:
setattr(component, "do_not_save_to_config", True)

# If changed, you need to change FaceSwapUnitSettings accordingly
# ORDER of parameters is IMPORTANT. It should match the result of FaceSwapUnitSettings
return gradio_components

0 comments on commit 2e718ee

Please sign in to comment.