Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

log a message on download

This commit is contained in:
Damian Stewart 2023-01-23 19:45:23 +01:00
parent 7a78ad0dfa
commit 6ee717c807
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/huggingface_downloader.py
View File

@ -37,6 +37,6 @@ def try_download_model_from_hf(repo_id: str,
downloaded_folder = huggingface_hub.snapshot_download(repo_id=repo_id, downloaded_folder = huggingface_hub.snapshot_download(repo_id=repo_id,
allow_patterns=allow_patterns, allow_patterns=allow_patterns,
ignore_patterns=ignore_patterns) ignore_patterns=ignore_patterns)
print(f"model with repo id {repo_id} downloaded to {downloaded_folder}")
is_sd1_attn, yaml_path = patch_unet(downloaded_folder) is_sd1_attn, yaml_path = patch_unet(downloaded_folder)
return downloaded_folder, is_sd1_attn, yaml_path return downloaded_folder, is_sd1_attn, yaml_path