Coool2 commited on
Commit
4c11be0
·
1 Parent(s): 51dd3c8

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
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, max_position_embeddings=1024)
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",