Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def sample_images(graphdef, state, x0, t):
|
|
| 35 |
o = flow(y, t[None])
|
| 36 |
return o
|
| 37 |
|
| 38 |
-
o = ode.odeint(flow_fn, x0, t, rtol=1e-
|
| 39 |
o = jnp.clip(o[-1], 0, 1)
|
| 40 |
return o
|
| 41 |
|
|
|
|
| 35 |
o = flow(y, t[None])
|
| 36 |
return o
|
| 37 |
|
| 38 |
+
o = ode.odeint(flow_fn, x0, t, rtol=1e-2)
|
| 39 |
o = jnp.clip(o[-1], 0, 1)
|
| 40 |
return o
|
| 41 |
|