red thing

[This](https://github.com/minimaxir/aitextgen/issues/87#issuecomment-770431967) appears to solve the problem?

Yes.

Also, as I said in a previous comment, replacing the dependencies section at the top of the notebook with the comment here: https://github.com/minimaxir/aitextgen/issues/87#issuecomment-770431967 resolves this issue. So it's probably a...

Looks like this is resolved!

just change it to ```py ai = aitextgen(model_folder=f"{run_dir}", to_gpu=True) ```

@channelz probably, yes.

The patch has been merged on huggingface! Is this feature planned in the near future?

> That's weird. Are you changing any other training settings? everything else is defaults. i tried again using a fresh copy of your notebook and 125M now works, but350M still...

> Having this issue too. If it matters, I'm using a pretty large text file (~20 MB) as the dataset, and I'm also getting this warning a short while after...

I have done this successfully using this script: https://github.com/huggingface/transformers/blob/master/src/transformers/models/gpt_neo/convert_gpt_neo_mesh_tf_to_pytorch.py