[model] add mistral small models (#6786)
Former-commit-id: e5e95c39bc4199fa89c67e34f9adaaa987058744
This commit is contained in:
@@ -183,8 +183,8 @@ class HuggingfaceEngine(BaseEngine):
|
||||
|
||||
if getattr(model.config, "model_type", None) in ["minicpmv", "minicpmo"]:
|
||||
gen_kwargs["input_ids"] = inputs
|
||||
del gen_kwargs["image_sizes"]
|
||||
gen_kwargs["tokenizer"] = tokenizer
|
||||
gen_kwargs.pop("image_sizes", None)
|
||||
|
||||
return gen_kwargs, prompt_length
|
||||
|
||||
|
||||
Reference in New Issue
Block a user