Method get_vector_field_from_eps

diff2flow/flow_obj.py:169–180

eps is the SD eps-parameterized vector field with the FM vector field is defined as z - eps First of all convert the x_t from the rectified flow trajectory to the original diffusion trajectory Then calculate the vector field from the eps-parameterized vector field

(self, noise, x_t, t)

        return vector_field
    
    def get_vector_field_from_eps(self, noise, x_t, t):
        """
        eps is the SD eps-parameterized vector field with
        the FM vector field is defined as z - eps

        First of all convert the x_t from the rectified flow trajectory to the original diffusion trajectory
        Then calculate the vector field from the eps-parameterized vector field
        """
        z_pred = self.predict_start_from_eps(x_t, t, noise)
        eps_pred = noise
        vector_field = z_pred - eps_pred                    # z - eps
        return vector_field
    
    def forward(self, x, t, **kwargs):
        """