update omost (#190)

* update omost
This commit is contained in:
ZhouTianchen
2024-09-09 17:39:46 +08:00
committed by GitHub
parent 1887885274
commit 995f3374f1
5 changed files with 55 additions and 29 deletions

View File

@@ -37,9 +37,9 @@ def tokenize_long_prompt(tokenizer, prompt, max_length=None):
class BasePrompter:
def __init__(self, refiners=[], extenders=[]):
self.refiners = refiners
self.extenders = extenders
def __init__(self):
self.refiners = []
self.extenders = []
def load_prompt_refiners(self, model_manager: ModelManager, refiner_classes=[]):