Peter Rosenberg

Results 2 comments of Peter Rosenberg

AttributeError: module 'jax.random' has no attribute 'KeyArray'

Ditto

Allocation on device | torch.OutOfMemoryError | SamplerCustomAdvanced | Flux.1

Similar experience - only when using xlabs flux nodes. (running on Colab Pro)