对于 transformer >= 4.56,请使用 `dtype` 而不是已淘汰的 `torch_dtype`
作者: xyf5432创建于 2026年8月28日更新于 2026年8月28日
在 `contrib/rlhflow/reward_modeling.py` 第 44 行, `AutoModelForSequenceClassification.from_pretrained` 使用 `torch_dtype=torch.bfloat16` 进行调用:
内容来源: OptimalScale/LMFlow