And save converted weights into safetensors instead of pickle
Co-authored-by: Benjamin Trom <benjamintrom@gmail.com>