OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 12:20:27 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Phaze A - Allow outputs divisible by 16

Browse Source
This commit is contained in:
torzdf 2023-09-01 09:50:47 +01:00
parent 5d00025884
commit 603fdc2a35
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/train/model/phaze_a.py
View File

@ -163,8 +163,8 @@ class Model(ModelBase):
""" """
def __init__(self, *args, **kwargs) -> None: def __init__(self, *args, **kwargs) -> None:
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
if self.config["output_size"] % 64 != 0: if self.config["output_size"] % 16 != 0:
raise FaceswapError("Phaze-A output shape must be a multiple of 64") raise FaceswapError("Phaze-A output shape must be a multiple of 16")
self._validate_encoder_architecture() self._validate_encoder_architecture()
self.config["freeze_layers"] = self._select_freeze_layers() self.config["freeze_layers"] = self._select_freeze_layers()

2
plugins/train/model/phaze_a_defaults.py
View File

@ -65,7 +65,7 @@ _DEFAULTS = {
"Resolution (in pixels) of the output image to generate.\n" "Resolution (in pixels) of the output image to generate.\n"
"BE AWARE Larger resolution will dramatically increase VRAM requirements."), "BE AWARE Larger resolution will dramatically increase VRAM requirements."),
"datatype": int, "datatype": int,
"rounding": 64, "rounding": 16,
"min_max": (64, 2048), "min_max": (64, 2048),
"group": "general", "group": "general",
"fixed": True}, "fixed": True},