Use HF hub checkpoint rather than local checkpoint as the default trained induction model.

Author: abhishekraok

README.md

@@ -149,7 +149,7 @@ python wandering_light/training/rl_grpo.py --batch-size 32 --model-name $SFT_OUT
 ### Evaluate local trained LLM 
 Assuming you have a checkpoint shown below.
 ```bash
-python wandering_light/evals/run_evaluation.py --budget=1 --eval_file=wandering_light/evals/data/random_inputs_500.py  --solver_names=[trained_local] --budget 1 --model-name checkpoints/saved/rl/long_sft_opt_125m_s35k_no_len/
+python wandering_light/evals/run_evaluation.py --budget=1 --eval_file=wandering_light/evals/data/random_inputs_500.py  --solver_names=[trained_local] --budget 1 --model-name abhishekraok/induction-basicfns-opt125m-longsft
 ```
 
 ### Evaluation Dashboard
@@ -164,5 +164,5 @@ First finetune it using SFT, using the `--task proposer` flag. Then evaluate it.
 
 ### Evaluate proposer
 ```bash
-python -m wandering_light.evals.evaluate_proposer --model checkpoints/saved/sft/proposer_opt_125m_2k --solver-model checkpoints/saved/rl/long_sft_opt_125m_s35k_no_len/ 
+python -m wandering_light.evals.evaluate_proposer --model checkpoints/saved/sft/proposer_opt_125m_2k --solver-model abhishekraok/induction-basicfns-opt125m-longsft 
 ```

wandering_light/constants.py

@@ -6,8 +6,7 @@
 
 # Evaluation file path
 DEFAULT_EVAL_FILE = "wandering_light/evals/data/random_inputs.py"
-# TODO Upload this to the internet
-DEFAULT_SOLVER_CHECKPOINT = "checkpoints/saved/rl/long_sft_opt_125m_s35k_no_len"
+DEFAULT_SOLVER_CHECKPOINT = "abhishekraok/induction-basicfns-opt125m-longsft"
 
 
 class Task(StrEnum):

wandering_light/web_ui/backend/main.py

@@ -81,7 +81,7 @@ class ExecutionResult(BaseModel):
 class SolverRequest(BaseModel):
     input_list: TypedListCreate
     output_list: TypedListCreate
-    checkpoint_path: str = "checkpoints/saved/rl/long_sft_opt_125m_s35k_no_len/"
+    checkpoint_path: str = "abhishekraok/induction-basicfns-opt125m-longsft"
 
 
 class SolverResponse(BaseModel):

wandering_light/web_ui/frontend/src/components/SolverPlayground.tsx

@@ -40,7 +40,7 @@ const SolverPlayground: React.FC = () => {
   const [inputItems, setInputItems] = useState('[1, 2, 3]');
   const [outputType, setOutputType] = useState('builtins.int');
   const [outputItems, setOutputItems] = useState('[2, 4, 6]');
-  const [checkpointPath, setCheckpointPath] = useState('../../../checkpoints/saved/rl/long_sft_opt_125m_s35k_no_len/');
+  const [checkpointPath, setCheckpointPath] = useState('abhishekraok/induction-basicfns-opt125m-longsft');
   const [solverResult, setSolverResult] = useState<SolverResponse | null>(null);
   const [isLoading, setIsLoading] = useState(false);
   const toast = useToast();