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",
|