Use small model in local

huridocs · Sep 27, 2024 · eb1469b · eb1469b
1 parent b603d0c
commit eb1469b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -18,9 +18,10 @@ services:
     container_name: worker_translations
     environment:
       LOGGING_LEVEL: "WARNING"
+      MODEL: "qwen:0.5b-text-v1.5-q2_K"
     init: true
     entrypoint: [ "python", "-m", "src.start_queue_processor" ]
     build:
       context: .
       dockerfile: Dockerfile
-    network_mode: host
+    network_mode: host
diff --git a/src/configuration.py b/src/configuration.py
@@ -10,7 +10,7 @@
 LANGUAGES_SHORT = ["en", "fr", "es", "ru", "ar", "sp"]
 LANGUAGES = ["English", "French", "Spanish", "Russian", "Arabic", "Spanish"]
 
-QUEUES_NAMES = os.environ.get("QUEUES_NAMES", "translation development_translation")
+QUEUES_NAMES = os.environ.get("QUEUES_NAMES", "translations development_translations")
 
 GRAYLOG_IP = os.environ.get("GRAYLOG_IP")
 REDIS_HOST = os.environ.get("REDIS_HOST", "127.0.0.1")

diff --git a/src/test_end_to_end.py b/src/test_end_to_end.py
@@ -13,7 +13,7 @@
 QUEUE = RedisSMQ(
     host=REDIS_HOST,
     port=REDIS_PORT,
-    qname="translation_tasks",
+    qname="translations_tasks",
     quiet=False,
 )
 
@@ -52,7 +52,7 @@ def get_results_message() -> TranslationResponseMessage:
             queue = RedisSMQ(
                 host=REDIS_HOST,
                 port=REDIS_PORT,
-                qname="translation_results",
+                qname="translations_results",
                 quiet=False,
             )
             message = queue.receiveMessage().exceptions(False).execute()