vision_unlearning.utils.model_management

Attributes

logger

Functions

save_model_card(repo_id, base_model, ...[, images, ...])

The resulting file looks like this: https://github.com/huggingface/hub-docs/blob/main/modelcard.md

Module Contents

vision_unlearning.utils.model_management.logger

vision_unlearning.utils.model_management.save_model_card(repo_id: str, base_model: str, dataset_forget_name: str, dataset_retain_name: str, repo_folder: str, images: Dict[str, PIL.Image.Image] = {}, eval_results: List[huggingface_hub.repocard_data.EvalResult] = [], tags: List[str] = [], hyperparameters: dict = {}, similarities_gr: List[float] = [], similarities_gf: List[float] = [])[source]

The resulting file looks like this: https://github.com/huggingface/hub-docs/blob/main/modelcard.md This looks hugginface-specific, so you may think it should be in integrations/huggingface.py, but it is actually a generic Readme saving

Adapted from The HuggingFace Inc. team. All rights reserved. Licensed under the Apache License, Version 2.0. Source: https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_lora.py