KOMUChat / model.py
ElPlaguister
WRAP Models
c90fdd5
raw
history blame
252 Bytes
class Model:
def __init__(self):
self.model = None
self.tokenizer = None
self.gen_config = None
self.INPUT_FORMAT = None
def generate(self, inputs:str) -> str:
outputs = inputs
return outputs