소스 검색

Update finetuning.py

Deleting comment
Jay Caldwell 3 달 전
부모
커밋
7e17d897fc
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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(