Add e5-mistral embedding model - step 3/3 (#988)
This commit is contained in:
@@ -223,6 +223,15 @@ def is_multimodal_model(model):
|
||||
raise ValueError("unrecognized type")
|
||||
|
||||
|
||||
def is_generation_model(model_architectures):
|
||||
if (
|
||||
"LlamaEmbeddingModel" in model_architectures
|
||||
or "MistralModel" in model_architectures
|
||||
):
|
||||
return False
|
||||
return True
|
||||
|
||||
|
||||
def decode_video_base64(video_base64):
|
||||
from PIL import Image
|
||||
|
||||
|
||||
Reference in New Issue
Block a user