vision_unlearning.utils.model_management ======================================== .. py:module:: vision_unlearning.utils.model_management Attributes ---------- .. autoapisummary:: vision_unlearning.utils.model_management.logger Functions --------- .. autoapisummary:: vision_unlearning.utils.model_management.save_model_card Module Contents --------------- .. py:data:: logger .. py:function:: 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] = []) 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