unable to train using autotrain-advance

by prajwalJumde - opened Dec 8, 2023

Dec 8, 2023

Getting error : "TypeError: StableLMEpochPreTrainedModel._set_gradient_checkpointing() got an unexpected keyword argument 'enable' "
when i try to use autotrain-advance with this model

rishiraj

Dec 10, 2023

for additional clarity, it is not an issue with AutoTrain Advance. Even using the SFTTrainer directly gives this error.

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