tf_retrieval_user_model / configuration.py
ashraq's picture
fix loading issue
6f3105b
from transformers import PretrainedConfig
class UserModelConfig(PretrainedConfig):
def __init__(self, **kwargs):
super().__init__(**kwargs)