Merge pull request #1 from isaac-mcfadyen/scheduler-fix

fix: switched to proper scheduler for fast/dev
2025-04-13 22:27:54 -04:00
parent 526a151f96 f51b4c9a16
commit 3c7c1b808c
1 changed files with 1 additions and 1 deletions
@@ -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,