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