comment out deepspeed for now

minimaxir · Mar 16, 2023 · 78a0bfd · 78a0bfd
1 parent f5ee331
commit 78a0bfd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py
@@ -11,7 +11,7 @@
 import pytorch_lightning as pl
 import torch
 from pkg_resources import resource_filename
-from pytorch_lightning.plugins import DeepSpeedPlugin
+# from pytorch_lightning.plugins import DeepSpeedPlugin
 from tqdm.auto import trange
 from transformers import (
     AutoConfig,