grf log prob keyword

homerjed · homerjed · commit 7e78b0b914a9 · 2025-06-04T16:18:25.000+02:00
diff --git a/.gitignore b/.gitignore
@@ -11,4 +11,5 @@ __unet.py
 cifar10.ipynb
 grfs.ipynb
 simple.ipynb
-mnist_clouds.py
+mnist_clouds.py
+ldm.py
diff --git a/examples/grfs.py b/examples/grfs.py
@@ -243,7 +243,7 @@ def diffuse(x, t, eps):
 key, key_L = jr.split(key)
 
 log_likelihood_fn = sbgm.ode.get_log_likelihood_fn(
-    model, sde, dataset.data_shape, exact_log_prob=True
+    model, sde, data_shape=dataset.data_shape, exact_log_prob=True
 )
 L_X = log_likelihood_fn(X[0], Q[0], A[0], key_L)
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "sbgm"
-version = "0.0.33"
+version = "0.0.34"
 description = "Score-based Diffusion models in JAX."
 readme = "README.md"
 requires-python ="~=3.12"

Original file line number	Diff line number	Diff line change
`@@ -243,7 +243,7 @@ def diffuse(x, t, eps):`
`243`	`243`	`key, key_L = jr.split(key)`
`244`	`244`
`245`	`245`	`log_likelihood_fn = sbgm.ode.get_log_likelihood_fn(`
`246`		`- model, sde, dataset.data_shape, exact_log_prob=True`
	`246`	`+ model, sde, data_shape=dataset.data_shape, exact_log_prob=True`
`247`	`247`	`)`
`248`	`248`	`L_X = log_likelihood_fn(X[0], Q[0], A[0], key_L)`
`249`	`249`