Update app.py
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ pipeline = FlaxStableDiffusionControlNetPipeline(
|
|
72 |
None,
|
73 |
dtype=jnp.float32,
|
74 |
)
|
75 |
-
controlnet_params = checkpoints.restore_checkpoint("
|
76 |
|
77 |
pipeline_params = {
|
78 |
"vae": vae_params,
|
|
|
72 |
None,
|
73 |
dtype=jnp.float32,
|
74 |
)
|
75 |
+
controlnet_params = checkpoints.restore_checkpoint("checkpoint_72001", target=None)
|
76 |
|
77 |
pipeline_params = {
|
78 |
"vae": vae_params,
|