Closed janfb closed 2 months ago
To Reproduce run test_correctness_of_batched_vs_seperate_sample_and_log_prob
test_correctness_of_batched_vs_seperate_sample_and_log_prob
Can you add some details about your environment? The tests all pass locally for me.
Sorry, I was a bit too early with the issue. It's not merged yet but only fails in #1153
To Reproduce run
test_correctness_of_batched_vs_seperate_sample_and_log_prob