File size: 277 Bytes
779c83e |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import PretrainedConfig
from typing import List
class RewardConfig(PretrainedConfig):
def __init__(
self,
base_model="EleutherAI/gpt-j-6b",
**kwargs,
):
self.base_model = base_model
super().__init__(**kwargs) |