fix: switch from pytorch_lightning to lightning

This commit is contained in:
Laurent Fainsin 2023-04-02 19:02:04 +02:00
parent e06a11bb49
commit 35c9b2630c

View file

@ -18,7 +18,7 @@ dependencies:
- torchvision
- torchaudio
- pytorch-cuda=11.8
- pytorch-lightning
- lightning
# deep learning libraries
- transformers
- datasets