kenlkehl commited on
Commit
b209fb2
·
verified ·
1 Parent(s): 39535dc

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +2 -2
config.py CHANGED
@@ -16,8 +16,8 @@ MODEL_CONFIG = {
16
  "embedder": "kenlkehl/mmai-trialspace", # e.g., "Qwen/Qwen3-Embedding-0.6B" or "./reranker_round2.model"
17
 
18
  # Large language model for patient history summarization
19
- #"llm": "kenlkehl/mmai-oncoreasoning-3b", # e.g., "microsoft/Phi-3-mini-4k-instruct" or "openai/gpt-oss-120b"
20
- "llm": "meta-llama/Llama-3.2-1B-Instruct",
21
 
22
  # ModernBERT classifier for eligibility prediction
23
  "trial_checker": "kenlkehl/mmai-trial-checker", # e.g., "answerdotai/ModernBERT-large" or "./modernbert-trial-checker"
 
16
  "embedder": "kenlkehl/mmai-trialspace", # e.g., "Qwen/Qwen3-Embedding-0.6B" or "./reranker_round2.model"
17
 
18
  # Large language model for patient history summarization
19
+ "llm": "kenlkehl/mmai-oncoreasoning-3b", # e.g., "microsoft/Phi-3-mini-4k-instruct" or "openai/gpt-oss-120b"
20
+ #"llm": "meta-llama/Llama-3.2-1B-Instruct",
21
 
22
  # ModernBERT classifier for eligibility prediction
23
  "trial_checker": "kenlkehl/mmai-trial-checker", # e.g., "answerdotai/ModernBERT-large" or "./modernbert-trial-checker"