Laurent/Diffusers.jl
1
0
Fork 0
mirror of https://github.com/Laurent2916/Diffusers.jl.git synced 2024-09-16 17:45:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

📝 (Schedulers) fix non matching docstrings

Browse source
This commit is contained in:
Laureηt 2023-07-29 15:51:13 +02:00
parent 4dfa3c92dc
commit 708e082874
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/Schedulers.jl
View file

@ -3,16 +3,14 @@ abstract type Scheduler end
"""
Add noise to clean data using the forward diffusion process.
cf. [[2006.11239] Denoising Diffusion Probabilistic Models](https://arxiv.org/abs/2006.11239) (Eq. 4)
## Input
* `scheduler::Scheduler`: scheduler to use
* `clean_data::AbstractArray`: clean data to add noise to
* `noise::AbstractArray`: noise to add to clean data
* `timesteps::AbstractArray`: timesteps used to weight the noise
* `x₀::AbstractArray`: clean data to add noise to
* `ϵ::AbstractArray`: noise to add to clean data
* `t::AbstractArray`: timesteps used to weight the noise
## Output
* `noisy_data::AbstractArray`: noisy data at the given timesteps
* `xₜ::AbstractArray`: noisy data at the given timesteps
"""
function add_noise(
scheduler::Scheduler,
@ -23,6 +21,8 @@ function add_noise(
⎷α̅ₜ = scheduler.⎷α̅[t]
⎷β̅ₜ = scheduler.⎷β̅[t]
# noisify clean data
# arxiv:2006.11239 Eq. 4
xₜ = ⎷α̅ₜ' .* x₀ + ⎷β̅ₜ' .* ϵ
return xₜ