File size: 317 Bytes
241f191 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
"""
"""
from models.base_model import Simulator
class MockSimulator(Simulator):
def __init__(self, *args, **kwargs):
pass
def strip_stoptokens(self, tokens):
return tokens
def generate(self, history, stream=True):
for text in ['hello', 'world']:
yield text, [11]
|