File size: 252 Bytes
22a2012 |
1 2 3 4 5 6 7 8 9 10 |
from transformers import PretrainedConfig
class HarmonConfig(PretrainedConfig):
model_type = "harmon"
def __init__(self, llm=None, mar=None, **kwargs):
super().__init__(**kwargs)
self.llm = llm
self.mar = mar
|