PixArt small fix

Issue #3 + some code changes from upstream.
2023-12-06 21:31:07 +01:00
parent 389c16f2f5
commit b31d8c517b
3 changed files with 7 additions and 3 deletions
@@ -38,7 +38,7 @@ def sample_pixart(model, seed, steps, cfg, noise_schedule, noise_schedule_vp, po
 	previewer = latent_preview.get_previewer(model.load_device, model.model.latent_format)

 	## Noise schedule.
-	betas = torch.tensor(gd.get_named_beta_schedule(noise_schedule, steps))
+	betas = torch.tensor(gd.get_named_beta_schedule(noise_schedule, 1000))
 	noise_schedule = NoiseScheduleVP(schedule=noise_schedule_vp, betas=betas)

 	## Convert your discrete-time `model` to the continuous-time