Browse Source

Update finetuning.py

Deleting comment
Jay Caldwell 2 months ago
parent
commit
7e17d897fc
1 changed files with 1 additions and 1 deletions
  1. 1 1
      src/llama_cookbook/finetuning.py

+ 1 - 1
src/llama_cookbook/finetuning.py

@@ -275,7 +275,7 @@ def main(**kwargs):
         elif torch.cuda.is_available():
             device_id = torch.cuda.current_device()
 
-        # This single line robustly sets use_orig_params for both freeze_LLM_only and PEFT (LoRA/QLoRA)
+       
         use_orig_params = train_config.freeze_LLM_only or train_config.use_peft
         
         model = FSDP(