sagemaker OOM

#3
by formfollows - opened

deploying this on sagemaker on a g6e.2x gives me a
torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 128.00 MiB. GPU 0 has a total capacity of 44.53 GiB

any setting i'm missing?

Sign up or log in to comment