Glen 7 months ago
parent
commit
4dd617ad37
2 changed files with 2 additions and 2 deletions
  1. 1 1
      .github/bench.py
  2. 1 1
      .github/workflows/build_and_test.yml

+ 1 - 1
.github/bench.py

@@ -19,7 +19,7 @@ async def measure_performance(api_endpoint: str, prompt: str) -> Dict[str, Any]:
     """
     """
     results: Dict[str, Any] = {}
     results: Dict[str, Any] = {}
     request_payload = {
     request_payload = {
-        "model": "llama-3.2-3b",
+        "model": "llama-3.2-1b",
         "messages": [{"role": "user", "content": prompt}],
         "messages": [{"role": "user", "content": prompt}],
         "temperature": 0,
         "temperature": 0,
         "stream": True
         "stream": True

+ 1 - 1
.github/workflows/build_and_test.yml

@@ -46,7 +46,7 @@ jobs:
           source env/bin/activate
           source env/bin/activate
           export PATH="/usr/local/bin:/opt/homebrew/bin:$PATH"
           export PATH="/usr/local/bin:/opt/homebrew/bin:$PATH"
           echo "${ALL_NODE_IDS}"
           echo "${ALL_NODE_IDS}"
-          DEBUG_DISCOVERY=7 DEBUG=7 exo --node-id="${MY_NODE_ID}" --node-id-filter="${ALL_NODE_IDS}" --chatgpt-api-port 52415 --disable-tui > output1.log 2>&1 &
+          DEBUG_DISCOVERY=7 DEBUG=7 exo --max-generate-tokens=100 --node-id="${MY_NODE_ID}" --node-id-filter="${ALL_NODE_IDS}" --chatgpt-api-port 52415 --disable-tui > output1.log 2>&1 &
           PID1=$!
           PID1=$!
           tail -f output1.log &
           tail -f output1.log &
           TAIL1=$!
           TAIL1=$!