Spaces:
Configuration error
Configuration error
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -210,7 +210,7 @@ def initialize_models(use_api_mode=False):
|
|
| 210 |
|
| 211 |
def __init__(self, model_name: str = "clip-ViT-B-32", **kwargs: Any) -> None:
|
| 212 |
super().__init__(**kwargs)
|
| 213 |
-
self._model = SentenceTransformer(model_name
|
| 214 |
|
| 215 |
@classmethod
|
| 216 |
def class_name(cls) -> str:
|
|
@@ -256,6 +256,7 @@ def initialize_models(use_api_mode=False):
|
|
| 256 |
|
| 257 |
|
| 258 |
embed_model = MultimodalCLIPEmbedding()
|
|
|
|
| 259 |
# Code LLM
|
| 260 |
code_llm = HuggingFaceLLM(
|
| 261 |
model_name="Qwen/Qwen2.5-Coder-3B-Instruct",
|
|
|
|
| 210 |
|
| 211 |
def __init__(self, model_name: str = "clip-ViT-B-32", **kwargs: Any) -> None:
|
| 212 |
super().__init__(**kwargs)
|
| 213 |
+
self._model = SentenceTransformer(model_name)
|
| 214 |
|
| 215 |
@classmethod
|
| 216 |
def class_name(cls) -> str:
|
|
|
|
| 256 |
|
| 257 |
|
| 258 |
embed_model = MultimodalCLIPEmbedding()
|
| 259 |
+
embed_model.max_seq_length = 1024
|
| 260 |
# Code LLM
|
| 261 |
code_llm = HuggingFaceLLM(
|
| 262 |
model_name="Qwen/Qwen2.5-Coder-3B-Instruct",
|