Yaroslav Aksenov

Error while converting checkpoints to Flax format

3

Directory with official LLaMa2 weights consists of `checklist.chk, consolidated.00.pth, params.json`. I want to use it to train CoH model and at first try to convert .pth model to Jax weights...

### System Info ```Shell accelerate==1.0.1 ``` ### Reproduction Since the latest versions split_batches is taken from DataLoaderConfiguration, but still can be passed as an argument to Accelerator constructor, which leads...