OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 00:20:09 +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:
super().__init__(*args, **kwargs)
if self.config["output_size"] % 64 != 0:
raise FaceswapError("Phaze-A output shape must be a multiple of 64")
if self.config["output_size"] % 16 != 0:
raise FaceswapError("Phaze-A output shape must be a multiple of 16")
self._validate_encoder_architecture()
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"
"BE AWARE Larger resolution will dramatically increase VRAM requirements."),
"datatype": int,
"rounding": 64,
"rounding": 16,
"min_max": (64, 2048),
"group": "general",
"fixed": True},