xai-org/grok-1 JAX version - Segmentation fault

by andy01
GNU/Linux ◆ xterm-256color ◆ bash 31 views

K8s pod (overlay FS)

nvidia driver 550.54.15

linux kernel 6.5.0-26-generic

https://github.com/xai-org/grok-1/issues/164

I suggest using PyTorch-based grok-1 version as described here:

https://github.com/xai-org/grok-1/issues/274#issuecomment-2015415953