From f51b4c9a1620143bdfad979b472be095f08d42e5 Mon Sep 17 00:00:00 2001 From: Isaac McFadyen Date: Fri, 11 Apr 2025 13:16:37 -0400 Subject: [PATCH] fix: switched to proper scheduler for fast/dev --- hdi1/nf4.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hdi1/nf4.py b/hdi1/nf4.py index bdf90cc..bb928cd 100644 --- a/hdi1/nf4.py +++ b/hdi1/nf4.py @@ -66,7 +66,7 @@ def load_models(model_type: str): pipe = HiDreamImagePipeline.from_pretrained( config["path"], - scheduler=FlowUniPCMultistepScheduler(num_train_timesteps=1000, shift=config["shift"], use_dynamic_shifting=False), + scheduler=config["scheduler"](num_train_timesteps=1000, shift=config["shift"], use_dynamic_shifting=False), tokenizer_4=tokenizer_4, text_encoder_4=text_encoder_4, torch_dtype=torch.bfloat16,