Add gh vars config for rewiew/ranking counts (#2479)

LAION-AI · Apr 12, 2023 · d984b6e · d984b6e
1 parent 734ff2c
commit d984b6e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/.github/workflows/deploy-to-node.yaml b/.github/workflows/deploy-to-node.yaml
@@ -50,6 +50,9 @@ jobs:
       LONELY_CHILDREN_COUNT: ${{ vars.LONELY_CHILDREN_COUNT }}
       P_LONELY_CHILD_EXTENSION: ${{ vars.P_LONELY_CHILD_EXTENSION }}
       P_ACTIVATE_BACKLOG_TREE: ${{ vars.P_ACTIVATE_BACKLOG_TREE }}
+      NUM_REQUIRED_RANKINGS: ${{ vars.NUM_REQUIRED_RANKINGS }}
+      NUM_REVIEWS_REPLY: ${{ vars.NUM_REVIEWS_REPLY }}
+      NUM_REVIEWS_INITIAL_PROMPT: ${{ vars.NUM_REVIEWS_INITIAL_PROMPT }}
       MIN_ACTIVE_RANKINGS_PER_LANG: ${{ vars.MIN_ACTIVE_RANKINGS_PER_LANG }}
       MAX_PROMPT_LOTTERY_WAITING: ${{ vars.MAX_PROMPT_LOTTERY_WAITING }}
       GOAL_TREE_SIZE: ${{ vars.GOAL_TREE_SIZE }}

diff --git a/ansible/deploy-to-node.yaml b/ansible/deploy-to-node.yaml
@@ -144,6 +144,15 @@
           TREE_MANAGER__P_ACTIVATE_BACKLOG_TREE:
             "{{ lookup('ansible.builtin.env', 'P_ACTIVATE_BACKLOG_TREE') |
             default('0.1', true) }}"
+          TREE_MANAGER__NUM_REQUIRED_RANKINGS:
+            "{{ lookup('ansible.builtin.env', 'NUM_REQUIRED_RANKINGS') |
+            default('3', true) }}"
+          TREE_MANAGER__NUM_REVIEWS_REPLY:
+            "{{ lookup('ansible.builtin.env', 'NUM_REVIEWS_REPLY') |
+            default('3', true) }}"
+          TREE_MANAGER__NUM_REVIEWS_INITIAL_PROMPT:
+            "{{ lookup('ansible.builtin.env', 'NUM_REVIEWS_INITIAL_PROMPT') |
+            default('3', true) }}"
           TREE_MANAGER__MIN_ACTIVE_RANKINGS_PER_LANG:
             "{{ lookup('ansible.builtin.env', 'MIN_ACTIVE_RANKINGS_PER_LANG') |
             default('0', true) }}"

diff --git a/docker/Dockerfile.backend b/docker/Dockerfile.backend
@@ -18,5 +18,6 @@ COPY ./backend/alembic.ini /app/alembic.ini
 COPY ./backend/main.py /app/main.py
 COPY ./backend/import.py /app/import.py
 COPY ./backend/export.py /app/export.py
+COPY ./backend/rerank.py /app/rerank.py
 COPY ./backend/oasst_backend /app/oasst_backend
 COPY ./backend/test_data /app/test_data