fix an unpinned dependency. bump version to 0.3.1

enjalot · Jul 5, 2024 · bd11c3e · bd11c3e
1 parent 120ce2c
commit bd11c3e
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,6 +4,7 @@ __pycache__
 venv
 testenv
 testenv-whl
+testenv-prod
 .env
 dist
 build

diff --git a/latentscope/__version__.py b/latentscope/__version__.py
@@ -1 +1 @@
-__version__ = '0.3.0'
+__version__ = '0.3.1'
diff --git a/latentscope/scripts/label_clusters.py b/latentscope/scripts/label_clusters.py
@@ -92,8 +92,8 @@ def labeler(dataset_id, text_column="text", cluster_id="cluster-001", model_id="
 The user will submit a bulleted list of items and you should choose a label that best summarizes the theme of the list so that someone browsing the labels will have a good idea of what is in the list. 
 Do not use punctuation, Do not explain yourself, respond with only a few words that summarize the list."""}
 
-    # TODO: why the extra 10 for openai?
-    max_tokens = model.params["max_tokens"] - len(enc.encode(system_prompt["content"])) - 10
+    # TODO: why the extra 50 for openai?
+    max_tokens = model.params["max_tokens"] - len(enc.encode(system_prompt["content"])) - 50
 
     # Create the lists of items we will send for summarization
     # Current looks like:

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 numpy~=1.26.4
-accelerate
+accelerate~=0.31.0
 aiohttp~=3.9.1
 aiolimiter~=1.1.0
 aiosignal~=1.3.1
-Original file line number
+Diff line change
@@ Expand Up / @@ -4,6 +4,7 @@ __pycache__ @@
     venv
     testenv
     testenv-whl
+    testenv-prod
     .env
     dist
     build
@@ Expand Down @@