migrate to new HF diffusers cache location

This commit is contained in:
Lincoln Stein
2023-03-05 08:20:24 -05:00
parent 7cf2f58513
commit ef8cf83b28
5 changed files with 25 additions and 43 deletions

View File

@ -640,7 +640,7 @@ def do_textual_inversion_training(
assert (
pretrained_model_name_or_path
), f"models.yaml error: neither 'repo_id' nor 'path' is defined for {model}"
pipeline_args = dict(cache_dir=global_cache_dir("diffusers"))
pipeline_args = dict(cache_dir=global_cache_dir("hub"))
# Load tokenizer
if tokenizer_name: