Merge pull request #290 from Haidra-Org/transformers-breaking-change

fix: prevent 'CLIPTokenizer' object has no attribute 'split_special_tokens' error
Haidra-Org · Aug 24, 2023 · 2d12545 · 2d12545
2 parents fc9946e + 0d085d3
commit 2d12545
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -2,6 +2,7 @@
 torch # We have to place it here otherwise it installs the CPU version
 pydantic==1.10.12
 horde_model_reference~=0.2.2
+transformers==4.30.2
 hordelib==1.6.2
 gradio
 pyyaml

diff --git a/worker/jobs/scribe.py b/worker/jobs/scribe.py
@@ -70,7 +70,7 @@ def start_job(self):
                     loop_retry += 1
                     time.sleep(3)
                     continue
-                if type(gen_req.json()) is not dict:
+                if not isinstance(gen_req.json(), dict):
                     logger.error(
                         (
                             f"KAI instance {self.bridge_data.kai_url} API unexpected response on generate: {gen_req}. "