AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

quality

Browse Source
This commit is contained in:
Nathan Lambert
2022-11-09 11:53:25 -08:00
parent 8fe2ff4b16
commit 56164f56fb
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/schedulers/scheduling_ddpm.py
View File

@@ -274,8 +274,6 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
new_config["predict_epsilon"] = predict_epsilon
self._internal_dict = FrozenDict(new_config)
t = timestep
if model_output.shape[1] == sample.shape[1] * 2 and self.variance_type in ["learned", "learned_range"]:
model_output, predicted_variance = torch.split(model_output, sample.shape[1], dim=1)
else: