minimaxir / ai-generated-magic-cards Goto Github PK

When downloading and setting up in the cell on the notebook (in a new copy) I`m getting the following error

`---------------------------------------------------------------------------
ImportError Traceback (most recent call last)
in
5 import sys
6 from jinja2 import Template
----> 7 from aitextgen import aitextgen
8
9 try:

2 frames
/usr/local/lib/python3.9/dist-packages/aitextgen/train.py in
12 import pytorch_lightning as pl
13 from pytorch_lightning.callbacks.progress import ProgressBarBase
---> 14 from pytorch_lightning.utilities import _TPU_AVAILABLE
15
16

ImportError: cannot import name '_TPU_AVAILABLE' from 'pytorch_lightning.utilities' (/usr/local/lib/python3.9/dist-packages/pytorch_lightning/utilities/init.py)

---

NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------`