From 73a0ca7ade7d02290da9d587125d467bcc5e29dd Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 20 Aug 2024 08:24:15 -0700 Subject: [PATCH] smaller epsilon when deriving flow from predict noise objective --- pyproject.toml | 2 +- rectified_flow_pytorch/rectified_flow.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 3648a73..7e625d4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "rectified-flow-pytorch" -version = "0.1.0" +version = "0.1.1" description = "Rectified Flow in Pytorch" authors = [ { name = "Phil Wang", email = "lucidrains@gmail.com" } diff --git a/rectified_flow_pytorch/rectified_flow.py b/rectified_flow_pytorch/rectified_flow.py index c488270..d8c4a96 100644 --- a/rectified_flow_pytorch/rectified_flow.py +++ b/rectified_flow_pytorch/rectified_flow.py @@ -263,7 +263,7 @@ def predict_flow(self, model: Module, noised, *, times): noise = output padded_times = append_dims(times, noised.ndim - 1) - flow = (noised - noise) / padded_times.clamp(min = 1e-2) + flow = (noised - noise) / padded_times.clamp(min = 1e-20) else: raise ValueError(f'unknown objective {self.predict}')