Chat-with-anyone-you-want-as-Gemma

Google Machine Learning Bootcamp를 진행하며 제작한 Gemma 활용 프로젝트.

프롬포트를 통한 아바타 대화, 파인튜닝 모델과 대화 두가지 기능 구현

• Project 폴더 : 실행하기 위한 파일들 수록
• Train 폴더 : 파인튜닝하기 위한 코드, 자료 수록

시연 이미지

시연 영상

1. 직접 입력 대화

custom.mp4

2. 파인 튜닝된 모델 대화

-.Clipchamp.mp4

핵심 로직

1. 프롬포팅 아바타

위 이미지와 같이 system을 통해 페르소나를 주고 대화에 임하도록 구현

2. 파인 튜닝 아바타

모델을 학습 후 허깅페이스에 올려서 사용.

학습 코드 및 데이터는 Train폴더 안에 기재되어있음.

제작 이슈

Gemma 2 의 경우 System role이 존재하지 않았는데, 이를 적용하기 위해 프롬포트 자체에 System role 을 적용. ( chat mod로는 되지 않음 )

참고 : https://huggingface.co/google/gemma-2-9b-it/discussions/15