From 0a03fbd544087466d771ddebc6216f9990722d2f Mon Sep 17 00:00:00 2001 From: ming tang Date: Mon, 26 Jan 2026 15:50:34 +0800 Subject: [PATCH] Change default model to HiDream-ai/HiDream-I1-Dev sd-legacy/stable-diffusion-v1-5 is not based on the DiT architecture and cannot run properly by default. Therefore, HiDream-ai/HiDream-I1-Dev is set as the default model. --- config/__init__.py | 2 +- core/__init__.py | 6 +++++- utils/__init__.py | 4 ++-- 3 files changed, 8 insertions(+), 4 deletions(-) diff --git a/config/__init__.py b/config/__init__.py index 55b304a..ce6b113 100644 --- a/config/__init__.py +++ b/config/__init__.py @@ -20,7 +20,7 @@ def init_config(args): device=args.device, served_model_name=args.served_model_name, steps_scale=args.steps_scale, - duplicate_scheduler=not args.disbale_duplicate_scheduler, + duplicate_scheduler=not args.disable_duplicate_scheduler, ) def get_config(): diff --git a/core/__init__.py b/core/__init__.py index 7ee0c55..2a655e3 100644 --- a/core/__init__.py +++ b/core/__init__.py @@ -34,6 +34,10 @@ def init_pipeline(args): _SHARED_PIPE = DiffusionPipeline.from_pretrained(args.model).to(args.device) else: _SHARED_PIPE = DiffusionPipeline.from_pretrained(args.model, device_map='balanced') - cache_dit.enable_cache(_SHARED_PIPE) + try: + cache_dit.enable_cache(_SHARED_PIPE) + except Exception as e: + print(f"[INFO] cache_dit not enabled (model not DiT?): {e}") + assert isinstance(_SHARED_PIPE, DiffusionPipeline), "pipeline init error, don't forget to assign pipeline to pipe var" diff --git a/utils/__init__.py b/utils/__init__.py index 1ff7590..92d670f 100644 --- a/utils/__init__.py +++ b/utils/__init__.py @@ -47,10 +47,10 @@ def parse_args(): help="Define the inference step benchmark. For hd, the benchmark value is 10. In the actual deployment, 10 is multiplied by this value and rounded up." ) parser.add_argument( - "--disbale-duplicate-scheduler", + "--disable-duplicate-scheduler", type=bool, default=False, - help="Whether to resolve https://github.com/huggingface/diffusers/issues/3672, fo HiDreams, use True", + help="Whether to resolve https://github.com/huggingface/diffusers/issues/3672, for HiDreams, use True", ) parser.add_argument( "--logging-level",