realfill-library/RealFill-Training-UI

1from __future__ import annotations2 3import pathlib4 5 6def find_exp_dirs(ignore_repo: bool = False) -> list[str]:7    repo_dir = pathlib.Path(__file__).parent8    exp_root_dir = repo_dir / 'experiments'9    if not exp_root_dir.exists():10        return []11    exp_dirs = sorted(exp_root_dir.glob('*'))12    exp_dirs = [13        exp_dir for exp_dir in exp_dirs14        if (exp_dir / 'model_index.json').exists()15    ]16    if ignore_repo:17        exp_dirs = [18            exp_dir for exp_dir in exp_dirs if not (exp_dir / '.git').exists()19        ]20    return [path.relative_to(repo_dir).as_posix() for path in exp_dirs]21 22 23def save_model_card(24    save_dir: pathlib.Path,25    base_model: str,26    target_image: str,27    target_mask: str,28) -> None:29    model_card = f'''---30license: creativeml-openrail-m31base_model: {base_model}32target_image: {target_image}33target_mask: {target_mask}34tags:35- stable-diffusion-inpainting36- stable-diffusion-inpainting-diffusers37- text-to-image38- diffusers39- realfill40inference: true41---42# RealFill - {save_dir.name}43 44These are RealFill weights for [{base_model}](https://huggingface.co/{base_model}). The weights were trained using [RealFill](https://realfill.github.io/).45'''46 47    with open(save_dir / 'README.md', 'w') as f:48        f.write(model_card)49