AnyRewardModel / README.md
XuyaoWang's picture
Update model files
d458a68 verified
|
raw
history blame
539 Bytes
# AnyRewardModel
## Usage
```python
from transformers import AutoModel, AutoProcessor
model = AutoModel.from_pretrained("PKU-Alignment/AnyRewardModel", trust_remote_code=True)
processor = AutoProcessor.from_pretrained("PKU-Alignment/AnyRewardModel", trust_remote_code=True)
```
## Note:
If you encounter the following error:
```
ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'
```
Please refer to guide at [blog](https://blog.csdn.net/lanxing147/article/details/136625264) for detailed resolution steps.