Source code for recwizard.modules.unicrs.configuration_unicrs_rec

from recwizard.configuration_utils import BaseConfig


[docs]class UnicrsRecConfig(BaseConfig):
[docs] def __init__(self, pretrained_model: str = "", kgprompt_config: dict = None, num_tokens: int = 0, pad_token_id: int = 0, **kwargs): super().__init__(**kwargs) self.pretrained_model = pretrained_model self.kgprompt_config = kgprompt_config self.num_tokens = num_tokens self.pad_token_id = pad_token_id