Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add option to override unet weights for conversion

Browse source
This commit is contained in:
Pierre Chapuis 2024-02-21 18:05:12 +01:00
parent 8f614e7647
commit d14c5bd5f8
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
scripts/conversion/convert_diffusers_unet.py
View file

@ -27,6 +27,9 @@ def setup_converter(args: Args) -> ModelConverter:
subfolder=args.subfolder, subfolder=args.subfolder,
low_cpu_mem_usage=False, low_cpu_mem_usage=False,
) )
if args.override_weights is not None:
sd = torch.load(args.override_weights) # type: ignore
source.load_state_dict(sd)
source_in_channels: int = source.config.in_channels # type: ignore source_in_channels: int = source.config.in_channels # type: ignore
source_clip_embedding_dim: int = source.config.cross_attention_dim # type: ignore source_clip_embedding_dim: int = source.config.cross_attention_dim # type: ignore
source_has_time_ids: bool = source.config.addition_embed_type == "text_time" # type: ignore source_has_time_ids: bool = source.config.addition_embed_type == "text_time" # type: ignore
@ -95,6 +98,15 @@ def main() -> None:
" runwayml/stable-diffusion-v1-5" " runwayml/stable-diffusion-v1-5"
), ),
) )
parser.add_argument(
"--override-weights",
type=str,
default=None,
help=(
"Path to a weights file to override the source model (keeping its config). "
"This is useful for models distributed as .pth files."
),
)
parser.add_argument( parser.add_argument(
"--to", "--to",
type=str, type=str,