Method get_vector_field_from_v

diff2flow/flow_obj.py:156–167

v is the SD v-parameterized vector field with v = sqrt(alpha_cumprod) * eps - sqrt(1 - alpha_cumprod) * z 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 th

(self, v, x_t, t)

        )
    
    def get_vector_field_from_v(self, v, x_t, t):
        """
        v is the SD v-parameterized vector field with v = sqrt(alpha_cumprod) * eps - sqrt(1 - alpha_cumprod) * z
        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 v-parameterized vector field
        """
        z_pred = self.predict_start_from_z_and_v(x_t, t, v)
        eps_pred = self.predict_eps_from_z_and_v(x_t, t, v)
        vector_field = z_pred - eps_pred                    # z - eps
        return vector_field
    
    def get_vector_field_from_eps(self, noise, x_t, t):
        """