@@ -23,7 +23,7 @@ class TestListHyperparametersInteg:
2323
2424 def test_sft_lora_returns_expected_params (self ):
2525 """SFT LORA returns a FineTuningOptions with known hyperparameters."""
26- hp = list_hyperparameters ("huggingface-llm-qwen2-5-7b -instruct" , "SFT" , "LORA" )
26+ hp = list_hyperparameters ("meta-textgeneration-llama-3-2-1b -instruct" , "SFT" , "LORA" )
2727
2828 assert isinstance (hp , FineTuningOptions )
2929 assert "learning_rate" in hp ._specs
@@ -33,15 +33,15 @@ def test_sft_lora_returns_expected_params(self):
3333
3434 def test_dpo_lora_has_additional_params (self ):
3535 """DPO LORA returns params including adam_beta (not present in SFT)."""
36- hp = list_hyperparameters ("huggingface-llm-qwen2-5-7b -instruct" , "DPO" , "LORA" )
36+ hp = list_hyperparameters ("meta-textgeneration-llama-3-2-1b -instruct" , "DPO" , "LORA" )
3737
3838 assert isinstance (hp , FineTuningOptions )
3939 assert "adam_beta" in hp ._specs
4040 assert "learning_rate" in hp ._specs
4141
4242 def test_rlvr_lora_returns_params (self ):
4343 """RLVR LORA returns FineTuningOptions with RL-specific params."""
44- hp = list_hyperparameters ("meta-textgeneration-llama-3-1-8b -instruct" , "RLVR" , "LORA" )
44+ hp = list_hyperparameters ("meta-textgeneration-llama-3-2-1b -instruct" , "RLVR" , "LORA" )
4545
4646 assert isinstance (hp , FineTuningOptions )
4747 assert "learning_rate" in hp ._specs
@@ -50,7 +50,7 @@ def test_rlvr_lora_returns_params(self):
5050 def test_accepts_enum_values (self ):
5151 """Accepts CustomizationTechnique and TrainingType enums."""
5252 hp = list_hyperparameters (
53- "huggingface-llm-qwen2-5-7b -instruct" ,
53+ "meta-textgeneration-llama-3-2-1b -instruct" ,
5454 CustomizationTechnique .SFT ,
5555 TrainingType .LORA ,
5656 )
@@ -60,7 +60,7 @@ def test_accepts_enum_values(self):
6060
6161 def test_get_info_does_not_raise (self ):
6262 """get_info() runs without error on returned object."""
63- hp = list_hyperparameters ("huggingface-llm-qwen2-5-7b -instruct" , "SFT" , "LORA" )
63+ hp = list_hyperparameters ("meta-textgeneration-llama-3-2-1b -instruct" , "SFT" , "LORA" )
6464 # Should print without raising
6565 hp .get_info ("learning_rate" )
6666
@@ -71,6 +71,6 @@ def test_invalid_model_raises(self):
7171
7272 def test_invalid_technique_raises (self ):
7373 """Technique not available for model raises an error."""
74- # PPO is not available on Qwen 2.5
74+ # PPO is not available on Llama 3.2
7575 with pytest .raises (Exception ):
76- list_hyperparameters ("huggingface-llm-qwen2-5-7b -instruct" , "PPO" , "LORA" )
76+ list_hyperparameters ("meta-textgeneration-llama-3-2-1b -instruct" , "PPO" , "LORA" )
0 commit comments