From 35c9b2630c98f2c42d24034f877ac91d785bbb3e Mon Sep 17 00:00:00 2001
From: Laurent Fainsin <laurent@fainsin.bzh>
Date: Sun, 2 Apr 2023 19:02:04 +0200
Subject: [PATCH] fix: switch from pytorch_lightning to lightning

---
 env.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/env.yml b/env.yml
index 99656eb..ad80029 100644
--- 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