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