Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/dome272/Diffusion-Models-pytorch
/ functions
Functions
37 in github.com/dome272/Diffusion-Models-pytorch
⨍
Functions
37
◇
Types & classes
9
↓ 6 callers
Method
__init__
(self, in_channels, out_channels, emb_dim=256)
modules.py:103
↓ 3 callers
Function
get_data
(args)
utils.py:24
↓ 3 callers
Function
save_images
(images, path, **kwargs)
utils.py:17
↓ 2 callers
Method
noise_images
(self, x, t)
ddpm.py:30
↓ 2 callers
Method
sample
(self, model, n, labels, cfg_scale=3)
ddpm_conditional.py:41
↓ 2 callers
Function
setup_logging
(run_name)
utils.py:36
↓ 1 callers
Function
launch
()
ddpm_conditional.py:112
↓ 1 callers
Function
launch
()
ddpm.py:94
↓ 1 callers
Method
noise_images
(self, x, t)
ddpm_conditional.py:32
↓ 1 callers
Function
plot_images
(images)
utils.py:9
↓ 1 callers
Method
pos_encoding
(self, t, channels)
modules.py:153
↓ 1 callers
Method
pos_encoding
(self, t, channels)
modules.py:217
↓ 1 callers
Method
prepare_noise_schedule
(self)
ddpm_conditional.py:29
↓ 1 callers
Method
prepare_noise_schedule
(self)
ddpm.py:27
↓ 1 callers
Method
reset_parameters
(self, ema_model, model)
modules.py:31
↓ 1 callers
Method
sample
(self, model, n)
ddpm.py:39
↓ 1 callers
Method
sample_timesteps
(self, n)
ddpm_conditional.py:38
↓ 1 callers
Method
sample_timesteps
(self, n)
ddpm.py:36
↓ 1 callers
Method
step_ema
(self, ema_model, model, step_start_ema=2000)
modules.py:23
↓ 1 callers
Function
train
(args)
ddpm_conditional.py:66
↓ 1 callers
Function
train
(args)
ddpm.py:61
↓ 1 callers
Method
update_average
(self, old, new)
modules.py:18
↓ 1 callers
Method
update_model_average
(self, ma_model, current_model)
modules.py:13
Method
__init__
(self, noise_steps=1000, beta_start=1e-4, beta_end=0.02, img_size=256, device="cuda")
ddpm_conditional.py:17
Method
__init__
(self, noise_steps=1000, beta_start=1e-4, beta_end=0.02, img_size=256, device="cuda")
ddpm.py:16
Method
__init__
(self, beta)
modules.py:8
Method
__init__
(self, channels, size)
modules.py:36
Method
__init__
(self, in_channels, out_channels, mid_channels=None, residual=False)
modules.py:59
Method
__init__
(self, in_channels, out_channels, emb_dim=256)
modules.py:80
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, device="cuda")
modules.py:129
Method
__init__
(self, c_in=3, c_out=3, time_dim=256, num_classes=None, device="cuda")
modules.py:190
Method
forward
(self, x)
modules.py:49
Method
forward
(self, x)
modules.py:72
Method
forward
(self, x, t)
modules.py:96
Method
forward
(self, x, skip_x, t)
modules.py:120
Method
forward
(self, x, t)
modules.py:163
Method
forward
(self, x, t, y)
modules.py:227