"docs/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "ab03dc4370acaaef05810465a077691472624b2b"
Unverified Commit 4b89aeff authored by Piyush Thakur's avatar Piyush Thakur Committed by GitHub
Browse files

[Type annotations] fixed in save_model_card (#6948)



fixed type annotations
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 0a1daade
...@@ -67,8 +67,8 @@ DATASET_NAME_MAPPING = { ...@@ -67,8 +67,8 @@ DATASET_NAME_MAPPING = {
def save_model_card( def save_model_card(
args, args,
repo_id: str, repo_id: str,
images=None, images: list = None,
repo_folder=None, repo_folder: str = None,
): ):
img_str = "" img_str = ""
if len(images) > 0: if len(images) > 0:
......
...@@ -56,7 +56,9 @@ check_min_version("0.27.0.dev0") ...@@ -56,7 +56,9 @@ check_min_version("0.27.0.dev0")
logger = get_logger(__name__, log_level="INFO") logger = get_logger(__name__, log_level="INFO")
def save_model_card(repo_id: str, images=None, base_model=str, dataset_name=str, repo_folder=None): def save_model_card(
repo_id: str, images: list = None, base_model: str = None, dataset_name: str = None, repo_folder: str = None
):
img_str = "" img_str = ""
for i, image in enumerate(images): for i, image in enumerate(images):
image.save(os.path.join(repo_folder, f"image_{i}.png")) image.save(os.path.join(repo_folder, f"image_{i}.png"))
......
...@@ -66,12 +66,12 @@ DATASET_NAME_MAPPING = { ...@@ -66,12 +66,12 @@ DATASET_NAME_MAPPING = {
def save_model_card( def save_model_card(
repo_id: str, repo_id: str,
images=None, images: list = None,
validation_prompt=None, validation_prompt: str = None,
base_model=str, base_model: str = None,
dataset_name=str, dataset_name: str = None,
repo_folder=None, repo_folder: str = None,
vae_path=None, vae_path: str = None,
): ):
img_str = "" img_str = ""
for i, image in enumerate(images): for i, image in enumerate(images):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment