File size: 305 Bytes
4962437
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from abc import ABC, abstractmethod

class AbstractModel(ABC):
    #abstract base class for language models
    def __init__():
        pass
    
    @abstractmethod
    def run(self, prompt):
        #generate text using language model
        pass

    def chat(self, prompt, history):
        pass