You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
from abc import ABC, abstractmethod
|
|
|
|
class AbstractModel(ABC):
|
|
#abstract base class for language models
|
|
|
|
@abstractmethod
|
|
def __init__(self, model_name **kwargs):
|
|
self.model_name = model_name
|
|
|
|
@abstractmethod
|
|
def generate(self, prompt):
|
|
#generate text using language model
|
|
pass
|
|
|
