Final_Assignment_Template

Configuration error

oremaz commited on Jun 15

Commit

51dd3c8

1 Parent(s): 08df1af

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -210,7 +210,7 @@ def initialize_models(use_api_mode=False):
                 def __init__(self, model_name: str = "clip-ViT-B-32", **kwargs: Any) -> None:
                     super().__init__(**kwargs)
-                    self._model = SentenceTransformer(model_name)
                 @classmethod
                 def class_name(cls) -> str:

                 def __init__(self, model_name: str = "clip-ViT-B-32", **kwargs: Any) -> None:
                     super().__init__(**kwargs)
+                    self._model = SentenceTransformer(model_name, max_position_embeddings=1024)
                 @classmethod
                 def class_name(cls) -> str: