Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-20 03:15:26 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

unet conversion: add option to skip init check

Browse source
This commit is contained in:
Pierre Chapuis 2024-01-16 18:40:43 +01:00
parent d2f38871fd
commit 7839c54ae8
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
scripts/conversion/convert_diffusers_unet.py
View file

@ -15,6 +15,7 @@ class Args(argparse.Namespace):
subfolder: str subfolder: str
half: bool half: bool
verbose: bool verbose: bool
skip_init_check: bool
def setup_converter(args: Args) -> ModelConverter: def setup_converter(args: Args) -> ModelConverter:
@ -49,7 +50,13 @@ def setup_converter(args: Args) -> ModelConverter:
"keyword": {"added_cond_kwargs": added_cond_kwargs} if source_has_time_ids else {}, "keyword": {"added_cond_kwargs": added_cond_kwargs} if source_has_time_ids else {},
} }
converter = ModelConverter(source_model=source, target_model=target, skip_output_check=True, verbose=args.verbose) converter = ModelConverter(
source_model=source,
target_model=target,
skip_init_check=args.skip_init_check,
skip_output_check=True,
verbose=args.verbose,
)
if not converter.run( if not converter.run(
source_args=source_args, source_args=source_args,
target_args=target_args, target_args=target_args,
@ -83,6 +90,11 @@ def main() -> None:
), ),
) )
parser.add_argument("--subfolder", type=str, default="unet", help="Subfolder. Default: unet.") parser.add_argument("--subfolder", type=str, default="unet", help="Subfolder. Default: unet.")
parser.add_argument(
"--skip-init-check",
action="store_true",
help="Skip check that source and target have the same layers count.",
)
parser.add_argument("--half", action="store_true", help="Convert to half precision.") parser.add_argument("--half", action="store_true", help="Convert to half precision.")
parser.add_argument( parser.add_argument(
"--verbose", "--verbose",