Merge pull request #49 from aws/revert-48-release-1.5.0

Revert "Release 1.5.0"

Author: zachgk

README.md

@@ -116,10 +116,6 @@ Nova Pro | Model Distillation for Post-Training | -   | -            | 1    | ml
 | DeepSeek R1 Distill Qwen 2 | LoRA | 32b    | 8192          | 2     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/deepseek/hf_deepseek_r1_distilled_qwen_32b_seq8k_gpu_lora.yaml) | [link](launcher_scripts/deepseek/run_hf_deepseek_r1_qwen_32b_seq8k_gpu_lora.sh) |
 | DeepSeek R1 Distill Qwen 2 | SFT  | 32b    | 16384         | 6     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/deepseek/hf_deepseek_r1_distilled_qwen_32b_seq16k_gpu_fine_tuning.yaml) | [link](launcher_scripts/deepseek/run_hf_deepseek_r1_qwen_32b_seq16k_gpu_fine_tuning.sh) |
 | DeepSeek R1 Distill Qwen 2 | LoRA | 32b    | 16384         | 2     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/deepseek/hf_deepseek_r1_distilled_qwen_32b_seq16k_gpu_lora.yaml) | [link](launcher_scripts/deepseek/run_hf_deepseek_r1_qwen_32b_seq16k_gpu_lora.sh) |
-| GPT OSS | LoRA   | 20b | 8192            | 1     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/gpt_oss/hf_gpt_oss_20b_seq8k_gpu_lora.yaml) | [link](launcher_scripts/gpt_oss/run_hf_gpt_oss_20b_seq8k_gpu_lora.sh) |
-| GPT OSS | LoRA   | 20b | 4096           | 1     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/gpt_oss/hf_gpt_oss_20b_seq4k_gpu_lora.yaml) | [link](launcher_scripts/gpt_oss/run_hf_gpt_oss_20b_seq4k_gpu_lora.sh) |
-| GPT OSS | LoRA   | 120b | 8192            | 8     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/gpt_oss/hf_gpt_oss_120b_seq8k_gpu_lora.yaml) | [link](launcher_scripts/gpt_oss/run_hf_gpt_oss_120b_seq8k_gpu_lora.sh) |
-| GPT OSS | LoRA   | 120b | 4096           | 4     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/gpt_oss/hf_gpt_oss_120b_seq4k_gpu_lora.yaml) | [link](launcher_scripts/gpt_oss/run_hf_gpt_oss_120b_seq4k_gpu_lora.sh) |
 | Llama 3.1 | QLoRA  | 405b | 131072          | 2     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/llama/hf_llama3_405b_seq128k_gpu_qlora.yaml) | [link](launcher_scripts/llama/run_hf_llama3_405b_seq128k_gpu_qlora.sh) |
 | Llama 3.1 | QLoRA  | 405b | 32768           | 2     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/llama/hf_llama3_405b_seq32k_gpu_qlora.yaml) | [link](launcher_scripts/llama/run_hf_llama3_405b_seq32k_gpu_qlora.sh) |
 | Llama 3.1 | LoRA   | 405b | 16384           | 6     | ml.p5.48xlarge    | GPU H100    | [link](recipes_collection/recipes/fine-tuning/llama/hf_llama3_405b_seq16k_gpu_lora.yaml) | [link](launcher_scripts/llama/run_hf_llama3_405b_seq16k_gpu_lora.sh) |

launcher/nemo/stages.py

@@ -303,9 +303,6 @@ def _make_launch_docker_container_text(self):
             if OmegaConf.select(self.cfg, "recipes.model.model_type", default=None) == "llama_v4":
                 transformers_upgrade_cmd = "pip install transformers==4.51.3"
                 post_launch_commands.append(transformers_upgrade_cmd)
-            if OmegaConf.select(self.cfg, "recipes.model.model_type", default=None) == "gpt_oss":
-                transformers_upgrade_cmd = "pip install transformers==4.55.0"
-                post_launch_commands.append(transformers_upgrade_cmd)
 
         launch_docker_container_text.append(f'  "{image}" sleep infinity')
         launch_docker_container_text.append("")
@@ -432,10 +429,6 @@ def _make_train_script_text(self, stage_cfg_path=None, port=41000) -> str:
                 transformers_upgrade_cmd = "pip install transformers==4.51.3"
                 script_text.append("")
                 script_text.append(transformers_upgrade_cmd)
-            if OmegaConf.select(self.cfg, "recipes.model.model_type", default=None) == "gpt_oss":
-                transformers_upgrade_cmd = "pip install transformers==4.55.0"
-                script_text.append("")
-                script_text.append(transformers_upgrade_cmd)
 
         script_text.append("")
         script_text.append(self._make_custom_call_string(stage_cfg_path))
@@ -775,9 +768,6 @@ def update_stage_specific_k8s_values(self, values_template):
         if OmegaConf.select(self.cfg, "recipes.model.model_type", default=None) == "llama_v4":
             transformers_upgrade_cmd = "pip install transformers==4.51.3"
             values_template.trainingConfig.pre_script.append(transformers_upgrade_cmd)
-        if OmegaConf.select(self.cfg, "recipes.model.model_type", default=None) == "gpt_oss":
-            transformers_upgrade_cmd = "pip install transformers==4.55.0"
-            values_template.trainingConfig.pre_script.append(transformers_upgrade_cmd)
 
         return values_template