error while trying import aitextgen

[dogukan33]

from aitextgen import aitextgen gives the error below...

Output exceeds the size limit. Open the full output data in a text editor

ImportError Traceback (most recent call last)
Cell In[7], line 1
----> 1 from aitextgen import aitextgen

File /opt/homebrew/lib/python3.10/site-packages/aitextgen/init.py:1
----> 1 from .aitextgen import aitextgen # noqa

File /opt/homebrew/lib/python3.10/site-packages/aitextgen/aitextgen.py:31
29 from .colab import create_gdrive_folder
30 from .TokenDataset import TokenDataset
---> 31 from .train import ATGProgressBar, ATGTransformer
32 from .utils import (
33 download_gpt2,
34 find_index_of_subset,
(...)
37 set_seed,
38 )
40 logger = logging.getLogger("aitextgen")

File /opt/homebrew/lib/python3.10/site-packages/aitextgen/train.py:14
12 import pytorch_lightning as pl
13 from pytorch_lightning.callbacks.progress import ProgressBarBase
---> 14 from pytorch_lightning.utilities import _TPU_AVAILABLE
17 class ATGTransformer(pl.LightningModule):
...
18 """
19 A training module for aitextgen.
20 """

ImportError: cannot import name '_TPU_AVAILABLE' from 'pytorch_lightning.utilities' (/opt/homebrew/lib/python3.10/site-packages/pytorch_lightning/utilities/init.py)

[pst2154]

Try to restart your kernal after installing

[2br-2b]

Duplicate of #200 - there's a solution also listed there