This commit is contained in:
wassname
2025-10-04 15:26:28 +08:00
parent 1fa4bb5ea4
commit 6810463fff
2 changed files with 5 additions and 6 deletions
+5 -5
View File
@@ -52,28 +52,28 @@ class Config:
num_seed: int = 8
max_iters: int = 950 # Small for demo; increase for more
n_shots: int = 16 # Number of in-context examples
model: str = "meta-llama/llama-3.1-8b-instruct" # Logprobs supported
model_id: str = "meta-llama/llama-3.1-8b-instruct" # Logprobs supported
provider_whitelist: Tuple[str] = None # None to let OpenRouter choose
out_dir: Path = Path("../outputs/icm") # Directory to save outputs
log_interval: int = 50 # Log progress every N iterations
C = Config(
model="qwen/qwen3-235b-a22b-2507",
model_id="qwen/qwen3-235b-a22b-2507",
provider_whitelist=[ 'Chutes','Nebius',],
)
C.out_dir.mkdir(parents=True, exist_ok=True)
# C = Config(
# model="qwen/qwen3-30b-a3b-instruct-2507",
# model_id="qwen/qwen3-30b-a3b-instruct-2507",
# provider_whitelist=[ 'Chutes','Nebius',],
# )
# C = Config(
# model="meta-llama/llama-3.1-70b-instruct",
# model_id="meta-llama/llama-3.1-70b-instruct",
# provider_whitelist=[ 'Cerebras','Nebius',],
# )
# C = Config(
# model="meta-llama/llama-3.1-8b-instruct",
# model_id="meta-llama/llama-3.1-8b-instruct",
# provider_whitelist=('Cerebras','Nebius',),
# )
-1
View File
@@ -9,7 +9,6 @@ authors = [
requires-python = ">=3.10"
dependencies = [
"adjusttext>=1.3.0",
"aiocache>=0.12.3",
"alembic>=1.16.5",
"altair>=5.5.0",
"anthropic>=0.69.0",