kaist-ai
/

janus-rm-7b

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

suehyunpark commited on May 29

Commit

85f5133

•

1 Parent(s): 8cce6c1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ Janus-RM-7B is a reward model created by training Janus-7B (which is trained for
 # Usage
 Here is example code to load the reward model and calculate a scalar reward on a model output.
-```
 from transformers import AutoConfig, AutoModel, AutoModelForCausalLM, AutoTokenizer
 import torch
 import torch.nn as nn

 # Usage
 Here is example code to load the reward model and calculate a scalar reward on a model output.
+```python
 from transformers import AutoConfig, AutoModel, AutoModelForCausalLM, AutoTokenizer
 import torch
 import torch.nn as nn