Skip to content

Commit d0c79be

Browse files
committed
default to ddim sampling, since not sure how to incorporate scale into logsnr yet
1 parent cabab0e commit d0c79be

File tree

3 files changed

+2
-3
lines changed

3 files changed

+2
-3
lines changed

README.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,6 @@ model = RIN(
4242
diffusion = GaussianDiffusion(
4343
model,
4444
image_size = 128,
45-
use_ddim = False,
4645
timesteps = 400,
4746
train_prob_self_cond = 0.9, # how often to self condition on latents
4847
scale = 1. # this will be set to < 1. for more noising and leads to better convergence when training on higher resolution images (512, 1024) - input noised images will be auto variance normalized

rin_pytorch/rin_pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -514,7 +514,7 @@ def __init__(
514514
*,
515515
image_size,
516516
timesteps = 1000,
517-
use_ddim = False,
517+
use_ddim = True,
518518
noise_schedule = 'sigmoid',
519519
objective = 'eps',
520520
schedule_kwargs: dict = dict(),

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
setup(
44
name = 'RIN-pytorch',
55
packages = find_packages(exclude=[]),
6-
version = '0.4.2',
6+
version = '0.4.3',
77
license='MIT',
88
description = 'RIN - Recurrent Interface Network - Pytorch',
99
author = 'Phil Wang',

0 commit comments

Comments
 (0)