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 - torchvision
- torchaudio - torchaudio
- pytorch-cuda=11.8 - pytorch-cuda=11.8
- pytorch-lightning - lightning
# deep learning libraries # deep learning libraries
- transformers - transformers
- datasets - datasets