DeepSpeedExamples
DeepSpeedExamples copied to clipboard
when I am running RLHF script, I encountered a error
My environments setting:
deepspeed==0.9.0,
torch==2.0.0+cu117
CUDA Version: 11.0
pretrained model is facebook/opt-350m
Who can help me solve this problem? Thanks
when I set enable_hybrid_engine=False, it's solved, what is the reason?
+1, same issue
+1