nbroad's picture
nbroad HF staff
change return_tensors to "np" for flax example
345863f