Spaces:
Sleeping
Sleeping
ping98k
commited on
Commit
·
41ec98d
1
Parent(s):
b241674
Update main.py
Browse files
main.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
from dotenv import load_dotenv
|
| 2 |
-
load_dotenv("./local.env")
|
| 3 |
import os, json, re, ast, gradio as gr
|
| 4 |
from concurrent.futures import ThreadPoolExecutor, as_completed
|
| 5 |
from tqdm import tqdm
|
|
@@ -30,7 +30,7 @@ class SimpleProgress:
|
|
| 30 |
return f"{self.prefix} {self.count}/{self.total} - ETA {eta}"
|
| 31 |
|
| 32 |
NUM_TOP_PICKS_DEFAULT = int(os.getenv("NUM_TOP_PICKS", 3))
|
| 33 |
-
POOL_SIZE_DEFAULT = int(os.getenv("POOL_SIZE",
|
| 34 |
MAX_WORKERS_DEFAULT = int(os.getenv("MAX_WORKERS", 100))
|
| 35 |
NUM_GENERATIONS_DEFAULT = int(os.getenv("NUM_GENERATIONS", 10))
|
| 36 |
API_BASE_DEFAULT = os.getenv("OPENAI_API_BASE", "")
|
|
@@ -40,9 +40,9 @@ PAIRWISE_FILTER_DEFAULT = os.getenv("ENABLE_PAIRWISE_FILTER", "true").lower() ==
|
|
| 40 |
GENERATE_MODEL_DEFAULT = os.getenv("GENERATE_MODEL", "gpt-4o-mini")
|
| 41 |
SCORE_MODEL_DEFAULT = os.getenv("SCORE_MODEL", "gpt-4o-mini")
|
| 42 |
PAIRWISE_MODEL_DEFAULT = os.getenv("PAIRWISE_MODEL", "gpt-4o-mini")
|
| 43 |
-
GENERATE_TEMPERATURE_DEFAULT = float(os.getenv("GENERATE_TEMPERATURE", "
|
| 44 |
-
SCORE_TEMPERATURE_DEFAULT = float(os.getenv("SCORE_TEMPERATURE", "0.
|
| 45 |
-
PAIRWISE_TEMPERATURE_DEFAULT = float(os.getenv("PAIRWISE_TEMPERATURE", "0.
|
| 46 |
SCORE_WITH_INSTRUCTION_DEFAULT = os.getenv("PASS_INSTRUCTION_TO_SCORE", "true").lower() == "true"
|
| 47 |
PAIRWISE_WITH_INSTRUCTION_DEFAULT = os.getenv("PASS_INSTRUCTION_TO_PAIRWISE", "true").lower() == "true"
|
| 48 |
CRITERIA_DEFAULT = "Factuality,Instruction Following,Precision"
|
|
|
|
| 1 |
from dotenv import load_dotenv
|
| 2 |
+
load_dotenv("./local.env",override=True)
|
| 3 |
import os, json, re, ast, gradio as gr
|
| 4 |
from concurrent.futures import ThreadPoolExecutor, as_completed
|
| 5 |
from tqdm import tqdm
|
|
|
|
| 30 |
return f"{self.prefix} {self.count}/{self.total} - ETA {eta}"
|
| 31 |
|
| 32 |
NUM_TOP_PICKS_DEFAULT = int(os.getenv("NUM_TOP_PICKS", 3))
|
| 33 |
+
POOL_SIZE_DEFAULT = int(os.getenv("POOL_SIZE", 6))
|
| 34 |
MAX_WORKERS_DEFAULT = int(os.getenv("MAX_WORKERS", 100))
|
| 35 |
NUM_GENERATIONS_DEFAULT = int(os.getenv("NUM_GENERATIONS", 10))
|
| 36 |
API_BASE_DEFAULT = os.getenv("OPENAI_API_BASE", "")
|
|
|
|
| 40 |
GENERATE_MODEL_DEFAULT = os.getenv("GENERATE_MODEL", "gpt-4o-mini")
|
| 41 |
SCORE_MODEL_DEFAULT = os.getenv("SCORE_MODEL", "gpt-4o-mini")
|
| 42 |
PAIRWISE_MODEL_DEFAULT = os.getenv("PAIRWISE_MODEL", "gpt-4o-mini")
|
| 43 |
+
GENERATE_TEMPERATURE_DEFAULT = float(os.getenv("GENERATE_TEMPERATURE", "0.9"))
|
| 44 |
+
SCORE_TEMPERATURE_DEFAULT = float(os.getenv("SCORE_TEMPERATURE", "0.6"))
|
| 45 |
+
PAIRWISE_TEMPERATURE_DEFAULT = float(os.getenv("PAIRWISE_TEMPERATURE", "0.6"))
|
| 46 |
SCORE_WITH_INSTRUCTION_DEFAULT = os.getenv("PASS_INSTRUCTION_TO_SCORE", "true").lower() == "true"
|
| 47 |
PAIRWISE_WITH_INSTRUCTION_DEFAULT = os.getenv("PASS_INSTRUCTION_TO_PAIRWISE", "true").lower() == "true"
|
| 48 |
CRITERIA_DEFAULT = "Factuality,Instruction Following,Precision"
|