Skip to content

Commit

Permalink
docs phrasing nit
Browse files Browse the repository at this point in the history
  • Loading branch information
Marton Havasi committed Dec 10, 2024
1 parent 06111bb commit f3fca1f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flow_matching/path/path.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ def sample(self, x_0: Tensor, x_1: Tensor, t: Tensor) -> PathSample:
def assert_sample_shape(self, x_0: Tensor, x_1: Tensor, t: Tensor):
assert (
t.ndim == 1
), f"The time vector t must be one of shape [batch_size]. Got {t.shape}."
), f"The time vector t must have shape [batch_size]. Got {t.shape}."
assert (
t.shape[0] == x_0.shape[0] == x_1.shape[0]
), f"Time t dimension must match the batch size [{x_1.shape[0]}]. Got {t.shape}"

0 comments on commit f3fca1f

Please sign in to comment.