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

🩹 move add_noise docstring in Schedulers.jl

Browse source
This commit is contained in:
Laureηt 2023-07-25 20:11:48 +02:00
parent b4ed6e3c99
commit 4122a47b6d
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/Schedulers.jl
View file

@ -1,23 +1,23 @@
abstract type Scheduler end
"""
Add noise to clean data using the forward diffusion process.
## Input
* scheduler (`Scheduler`): scheduler object.
* 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.
## Output
* noisy_data (`AbstractArray`): noisy data at the given timesteps.
"""
function add_noise(
scheduler::Scheduler,
clean_data::AbstractArray,
noise::AbstractArray,
timesteps::AbstractArray,
)
"""
Add noise to clean data using the forward diffusion process.
Args:
scheduler (`Scheduler`): scheduler object.
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.
Returns:
`AbstractArray`: noisy data at the given timesteps.
"""
sqrt_α_cumprod_t = scheduler.sqrt_α_cumprods[timesteps]
sqrt_one_minus_α_cumprod_t = scheduler.sqrt_one_minus_α_cumprods[timesteps]