sagemaker OOM

by formfollows - opened Mar 12

Mar 12

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?

cdm233

Mar 21

OOM during peft finetuning? Can you provide more context?

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment