Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • dg845's avatar
    Improve LCM(-LoRA) Distillation Scripts (#6420) · f3d1333e
    dg845 authored 
    * Make WDS pipeline interpolation type configurable.

* Make the VAE encoding batch size configurable.

* Make lora_alpha and lora_dropout configurable for LCM LoRA scripts.

* Generalize scalings_for_boundary_conditions function and make the timestep scaling configurable.

* Make LoRA target modules configurable for LCM-LoRA scripts.

* Move resolve_interpolation_mode to src/diffusers/training_utils.py and make interpolation type configurable in non-WDS script.

* apply suggestions from review
    f3d1333e
training_utils.py 16.7 KB