fix: switch from pytorch_lightning to lightning

2023-04-02 19:02:04 +02:00 · 2023-04-02 19:02:04 +02:00 · 35c9b2630c
parent e06a11bb49
commit 35c9b2630c
1 changed files with 1 additions and 1 deletions
--- a/env.yml
+++ b/env.yml
@ -18,7 +18,7 @@ dependencies:
  - torchvision
  - torchaudio
  - pytorch-cuda=11.8
-  - pytorch-lightning
+  - lightning
  # deep learning libraries
  - transformers
  - datasets