theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-03-14 15:57:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Tree: Format

Browse Source
This commit is contained in:
turboderp
2025-06-15 19:30:38 +02:00
parent 1c9891bf04
commit 21c5af48e1
2 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
common/multimodal.py
View File

@@ -22,14 +22,12 @@ class MultimodalEmbeddingWrapper(BaseModel):
async def add(self, url: str):
# Determine the type of vision embedding to use
if not self.type:
if (
dependencies.exllamav2 and
isinstance(model.container.vision_model, ExLlamaV2VisionTower)
if dependencies.exllamav2 and isinstance(
model.container.vision_model, ExLlamaV2VisionTower
):
self.type = "ExLlamaV2MMEmbedding"
elif (
dependencies.exllamav3 and
isinstance(model.container.vision_model, Model)
elif dependencies.exllamav3 and isinstance(
model.container.vision_model, Model
):
self.type = "MMEmbedding"
@@ -43,4 +41,4 @@ class MultimodalEmbeddingWrapper(BaseModel):
self.content.append(embedding)
self.text_alias.append(embedding.text_alias)
else:
logger.error("No valid vision model to create embedding")
logger.error("No valid vision model to create embedding")