AttributeError: 'UnspecifiedValue' object has no attribute '_parsed_pspec'

[tralala87]

While running:

pipeline, params = FlaxStableDiffusionWalkPipeline.from_pretrained

I get:

AttributeError Traceback (most recent call last)
/usr/local/lib/python3.8/dist-packages/jax/_src/prng.py in threefry_seed(seed)
824 # TODO(jakevdp): in X64 mode, this can generate 64-bit computations for 32-bit
825 # inputs. We should avoid this.
--> 826 k2 = convert(jnp.bitwise_and(seed, np.uint32(0xFFFFFFFF)))
827 return lax.concatenate([k1, k2], 0)
828

AttributeError: 'UnspecifiedValue' object has no attribute '_parsed_pspec'

[Atomic-Germ]

Could you please give your environment, and in what way you were running that?

[tralala87]

Google Colab Pro TPU