ping98k commited on
Commit
41ec98d
·
1 Parent(s): b241674

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -5
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", 5))
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", "1.0"))
44
- SCORE_TEMPERATURE_DEFAULT = float(os.getenv("SCORE_TEMPERATURE", "0.1"))
45
- PAIRWISE_TEMPERATURE_DEFAULT = float(os.getenv("PAIRWISE_TEMPERATURE", "0.1"))
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"