Yaowei Zheng
Yaowei Zheng
@bigcash 已更新,感谢提醒
Could you run it again on a single GPU by setting `CUDA_VISIBLE_DEVICES=0`?
用 accelerate launch src/train.py
目前还没支持,之后有空加一下
predict 可能不支持 deepspeed
cc @yzoaim please add deepspeed zero3 + batch predict to document
cc @BUAADreamer
应该是存在的https://github.com/huggingface/transformers/blob/v4.49.0/src/transformers/models/llava_next/processing_llava_next.py#L152-L153
cc: @mlinmg