hy3 set cache dir

2025-10-30 03:02:30 +03:00 · 2025-10-30 03:02:30 +03:00 · 8049936e66
parent f7fef1f5b5
commit 8049936e66
1 changed files with 1 additions and 0 deletions
--- a/pipelines/model_hyimage.py
+++ b/pipelines/model_hyimage.py
@ -53,6 +53,7 @@ def load_hyimage3(checkpoint_info, diffusers_load_config={}): # pylint: disable=
    load_args, quant_args = model_quant.get_dit_args(diffusers_load_config, module='Model', device_map=True, allow_quant=allow_quant)
    pipe = transformers.AutoModelForCausalLM.from_pretrained(
        repo_id,
+        cache_dir=shared.opts.diffusers_dir,
        trust_remote_code=True,
        attn_implementation="sdpa",
        moe_impl="eager",