본문 바로가기
AI/GPT

[ChatGPT API] 챗지피티 fine-tune 및 API 사용법

by Mujae98 2024. 1. 17.

ChatGPT API를 사용하려면 OpenAI 메인페이지에서 API를 선택하고 아래처럼 API-keys에 가서 api-key를 발급 받아야한다.

 

그리고 아래와 같은 코드로 사용하면 됨.

from openai import OpenAI
client = OpenAI()

completion = client.chat.completions.create(
  model="gpt-3.5-turbo",#사용하고 싶은 model 종류 -> 토큰제한과 얼마인지 확인하고 고르자.
  messages=[
    {"role": "system", "content": "전체적인 대화 맥락"},
    {"role": "user", "content": "여기다 행동을 요청하면 됨"}
  ]
)

print(completion.choices[0].message)

 

참쉽죠잉??

 

fine-tune은 대답의 톤이나 형식 같은 질적인 부분을 향상시키고 싶거나, 뭔가 더 자신만의 모델로 만들고 싶을 때 사용하면 된다.

토큰수를 줄일 수 있긴하나, 학습시킬 때 돈이 드니 잘 생각하고 쓰셨으면 한다.

fine-tune 방법 자체는 상당히 쉽다. train 데이터파일을 다음과 같이 형식만 맞춰주고 upload하면 된다.

{"messages": [{"role": "system", "content": "원하는 맥락, 형식 등등"}, {"role": "user", "content": 명령"}, {"role": "assistant", "content": "원하는 대답(정답)"}]}