| """ | |
| """ | |
| 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] | |