diff --git a/docker/requirements-cpu.txt b/docker/requirements-cpu.txt index 86fad239d93df3e2c70be7113b487e4a18898377..f08054340ac25eabd1dbf8b88c2d91d4795accab 100644 --- a/docker/requirements-cpu.txt +++ b/docker/requirements-cpu.txt @@ -19,6 +19,7 @@ torchtext torchdata lightning + tensorboard keras transformers numpy diff --git a/docker/requirements-gpu.txt b/docker/requirements-gpu.txt index 15808175bab0224a0efc1876abdf840eeb8fbf13..fe31bda2b1152794b81b8a1ae0ea04f1a90a80ac 100644 --- a/docker/requirements-gpu.txt +++ b/docker/requirements-gpu.txt @@ -18,6 +18,7 @@ torchtext torchdata lightning + tensorboard keras transformers numpy