Skip to content

Commit cb3d345

Browse files
committed
Fixing bridge simulation
1 parent 95b5492 commit cb3d345

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

language_modeling_via_stochastic_processes/transformers/examples/pytorch/text-generation/run_decoding_from_embeddings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def simulate_brownian_bridge(B_0, B_T, num_samples, sentence_lengths, dt=0.05, m
8888
x_tp1 = x_t * (1- dt/(1-t)) + (dt/(1-t))*B_T + noise
8989
length_idx = step % len(sentence_lengths)
9090
bridge += [x_tp1] * sentence_lengths[length_idx]
91-
x_tp1 = x_t
91+
x_t = x_tp1
9292

9393
length_idx = step % len(sentence_lengths)
9494
bridge += [B_T] * sentence_lengths[length_idx]

0 commit comments

Comments
 (0)