OOM with lora and half precision, but it works with full precision #18453
Unanswered
tshu-w
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
I encountered an OOM error when using peft and half-precision, but training works fine with full precision. I believe there is an issue here, but I don't have time to complete the reproducible code. Therefore, I started this discussion for now.
Beta Was this translation helpful? Give feedback.
All reactions