diffusers
f0c81562 - Add `final_sigma_zero` to UniPCMultistep (#7517)

Comment changes are shownComment changes are hidden
Commit
1 year ago
Add `final_sigma_zero` to UniPCMultistep (#7517) * Add `final_sigma_zero` to UniPCMultistep Effectively the same trick as DDIM's `set_alpha_to_one` and DPM's `final_sigma_type='zero'`. Currently False by default but maybe this should be True? * `final_sigma_zero: bool` -> `final_sigmas_type: str` Should 1:1 match DPM Multistep now. * Set `final_sigmas_type='sigma_min'` in UniPC UTs
Author
Parents
  • src/diffusers/schedulers
    • File
      scheduling_unipc_multistep.py
  • tests/schedulers
    • File
      test_scheduler_unipc.py