메인 콘텐츠로 건너뛰기
Serverless Inference를 사용하면 일부 기본 모델에서 커스텀 LoRA를 사용할 수 있습니다. 이 튜토리얼에서는 TRL 라이브러리를 사용해 지도 사후학습으로 파인튜닝된 LoRA를 만든 다음, Serverless Inference API 또는 플레이그라운드에서 사용할 수 있도록 W&B에 아티팩트로 업로드하는 과정을 안내합니다. 이 예시에서는 쿼리/Response 쌍으로 이루어진 데이터셋을 사용해 모델이 카우보이처럼 응답하도록 파인튜닝하지만, 같은 워크플로를 어떤 캐릭터나 작업에도 맞게 적용할 수 있습니다. 이 튜토리얼은 자체 추론 인프라를 관리하지 않고도 기본 모델의 동작을 사용자 지정하려는 개발자를 위한 것입니다. 서버리스 강화학습을 제공하는 Serverless RL을 사용해 LoRA를 만들 수도 있습니다. 이 튜토리얼은 세 부분으로 구성됩니다. 사후학습 데이터셋을 준비하고, LoRA를 생성하고 업로드하는 사후학습 스크립트를 실행한 다음, 완성된 LoRA를 플레이그라운드 또는 코드에서 사용해 봅니다.”

사후학습(Post-training) 데이터셋

이 섹션에서는 모델을 미세 조정하는 데 사용되는 예시 데이터셋을 제공합니다. 다음 데이터셋에는 메시지 목록 형식의 50개 쿼리-Response 쌍이 포함되어 있습니다. 예시는 다음과 같습니다:
사용자: “What is your favorite color?”
어시스턴트: “Well, pardner, my favorite color’s the blazin’ orange of a desert sunset.”
예시 파일에는 각 줄마다 JSON 객체 하나가 들어 있습니다. 다음 데이터를 작업 디렉터리에 cowboy_examples.jsonl로 저장하세요.
cowboy_examples.jsonl
이 데이터셋을 저장하면 사후학습(Post-training)을 실행할 준비가 됩니다.

사후학습(Post-training)

이 섹션에서는 데이터셋에서 LoRA 어댑터를 생성하고 이를 W&B에 업로드하는 트레이닝 스크립트를 실행하는 방법을 안내합니다. 이 스크립트는 JSONL 파일의 예시를 사용해 LoRA 어댑터를 트레이닝한 다음, Serverless Inference API 또는 플레이그라운드에서 사용할 수 있도록 W&B에 아티팩트로 업로드합니다. 전체적으로 이 스크립트는 다음을 수행합니다:
  1. W&B에 로그인합니다. W&B Models의 Hugging Face Transformers 인테그레이션은 트레이닝 진행 상황과 메트릭을 자동으로 기록합니다.
  2. Hugging Face에서 기본 모델(OpenPipe/Qwen3-14B-Instruct)을 로드합니다.
  3. 파일 상단 근처에 스크립트가 상수로 정의한 rank와 alpha 등의 하이퍼파라미터를 사용해 LoRA를 설정합니다.
  4. 파일의 예시를 데이터셋으로 로드한 다음 SFTTrainer를 실행합니다. 기본적으로 스크립트는 모든 예시를 사용합니다.
  5. LoRA를 저장하고, Serverless Inference에서 사용할 수 있도록 W&B에 아티팩트로 업로드합니다.
스크립트 실행이 끝나면 브라우저에서 마지막으로 출력된 URL을 열어 저장된 아티팩트를 확인하세요. 다음과 같은 형태입니다: Artifact URL: https://wandb.ai/[YOUR-ENTITY]/create-lora-tutorial/artifacts/lora/OpenPipe_Qwen3-14B-Instruct_cowboy/v0 다음 프로그램을 create_lora.py로 저장하고, ENTITY 값을 자신의 W&B entity로 업데이트하세요. 이 스크립트는 의존성을 선언하는 데 inline script metadata를 사용하므로, 별도의 가상 환경을 관리하지 않고도 uv로 직접 실행할 수 있습니다.
create_lora.py
uv를 사용해 스크립트를 실행하세요:
실행 시간은 하드웨어에 따라 달라집니다. 트레이닝 속도를 높이기 위해 --max-examples=10 인수를 추가할 수 있지만, 예시 수가 적을수록 LLM이 캐릭터를 유지하며 응답하는 정도가 줄어듭니다. 스크립트가 완료되면 학습된 LoRA 어댑터가 W&B 아티팩트로 저장되며, Serverless Inference에서 사용할 준비가 됩니다.

LoRA 사용하기

이 섹션에서는 생성한 LoRA를 플레이그라운드에서 대화형으로 또는 프로그래밍 방식으로 사용해 보는 방법을 설명합니다. 생성한 LoRA를 W&B Weave 플레이그라운드에서 사용해 볼 수 있습니다. 아티팩트 URL을 열고 Try in playground를 클릭하세요.
Artifacts UI의 LoRA
그런 다음 채팅 인터페이스 하단에 프롬프트를 입력하세요.
플레이그라운드 UI의 LoRA
코드에서 LoRA를 사용하려면 단계별 안내가 포함된 Serverless LoRA Inference 사용 가이드를 참고하세요.

다음 step

이제 작동하는 LoRA가 있으므로, 트레이닝 선택이 결과에 어떤 영향을 미치는지 확인하기 위해 더 실험해 볼 수 있습니다:
  • 더 적은 수의 예시로 LoRA를 트레이닝해도 여전히 원하는 효과가 나는지 확인해 보세요.
  • 데이터셋의 Response를 변경하여 해적이나 닌자 같은 다른 캐릭터를 보여주도록 해 보세요.