search

alembics / disco-diffusion

Other
7.46k stars 1.13k forks source link

An easy fix... init_scale vs args.init_scale #79

Closed aztecman closed 2 years ago

aztecman commented 2 years ago

if init is not None and args.init_scale: init_losses = lpips_model(x_in, init) loss = loss + init_losses.sum() * args.init_scale

instead should be

if init is not None and init_scale: init_losses = lpips_model(x_in, init) loss = loss + init_losses.sum() * init_scale

because earlier we might have set init_scale to be equal to frame_scale (if we are doing animation)

otherwise, init_scale is just a dead end variable

aletts commented 2 years ago

Thanks. I've just applied the change.