From f11ddf88e42f2171eebced18aaba304adc63208e Mon Sep 17 00:00:00 2001
From: gonzalomartingarcia0 <gonzalomartingarcia0@gmail.com>
Date: Sat, 22 Jul 2023 12:30:18 +0200
Subject: [PATCH] clipping added to combat class-free guidence lerp
---
evaluation/sample.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/evaluation/sample.py b/evaluation/sample.py
index a7a6e3b..4d98deb 100644
--- a/evaluation/sample.py
+++ b/evaluation/sample.py
@@ -75,6 +75,8 @@ def cdm_sampler(model, checkpoint, experiment_path, device, intermediate=False,
y = torch.randint(0, 3, (batch_size,)).to(device)
# generate images
generated = model.sample(y=y, batch_size=y.size(0))
+ # clip the values to between -1 and 1
+ generated = generated.clamp(-1, 1)
# save images
for i in range(generated.size(0)):
image = back2pil(generated[i])
--
GitLab