huggingface_hub deprecate resume_download

This commit is contained in:
Qing
2024-11-23 20:14:28 +08:00
parent 341f610159
commit 683298dd5f
2 changed files with 3 additions and 6 deletions

View File

@@ -36,10 +36,7 @@ def cli_download_model(model: str):
from diffusers import DiffusionPipeline
downloaded_path = handle_from_pretrained_exceptions(
DiffusionPipeline.download,
pretrained_model_name=model,
variant="fp16",
resume_download=True,
DiffusionPipeline.download, pretrained_model_name=model, variant="fp16"
)
logger.info(f"Done. Downloaded to {downloaded_path}")