Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix new k-diffusers

Browse Source
This commit is contained in:
Vladimir Mandic
2023-09-26 08:53:46 -04:00
parent 6650602289
commit 38402c8e6e
6 changed files with 29 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/modelloader.py
+15 -1
View File
@@ -174,10 +174,24 @@ def download_diffusers_model(hub_id: str, cache_dir: str = None, download_config
shared.log.debug(f"Diffusers authentication: {token}")
hf.login(token)
pipeline_dir = None
ok = True
try:
pipeline_dir = DiffusionPipeline.download(hub_id, **download_config)
except Exception as e:
shared.log.error(f"Diffusers download error: {hub_id} {e}")
ok = False
shared.log.warning(f"Diffusers download error: {hub_id} {e}")
if not ok:
try:
download_config.pop('load_connected_pipeline')
download_config.pop('variant')
pipeline_dir = hf.snapshot_download(hub_id, **download_config)
except Exception as e:
shared.log.warning(f"Diffusers hub download error: {hub_id} {e}")
if pipeline_dir is None:
shared.log.error(f"Diffusers no pipeline folder: {hub_id}")
return
try:
model_info_dict = hf.model_info(hub_id).cardData if pipeline_dir is not None else None # pylint: disable=no-member # TODO Diffusers is this real error?
except Exception: