from_pt after all

app.py

@@ -33,6 +33,7 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         controlnet=cnet,
         revision="flax",
         dtype=jnp.bfloat16,
+        from_pt=True
         )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()