Dependence on Apex canceled.

rc-zb · Apr 18, 2024 · 865a091 · 865a091
1 parent ea35b49
commit 865a091
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+# Python #
+__pycache__/
diff --git a/train_helpers.py b/train_helpers.py
@@ -15,7 +15,9 @@
 from data import mkdir_p
 from contextlib import contextmanager
 import torch.distributed as dist
-from apex.optimizers import FusedAdam as AdamW
+# from apex.optimizers import FusedAdam as AdamW
+#👆 Nvidia Apex can't be easily deployed. This commenting would block VDVAE from running properly.
+#   However, it should not interfere Brain-Diffuser.
 from vae import VAE
 from torch.nn.parallel.distributed import DistributedDataParallel