Skip to content

2024 프로메테우스 해커톤 본선 진출작: AI 모델링 및 인프런스 서버 구축

Notifications You must be signed in to change notification settings

dhdbsrlw/eternal-tales-AI

Repository files navigation

AI Modelling ( Multi-modal Chatbot )

1. Task

Real-time Text and Image generation corresponding to user's message

2. Model

2.1. Text Generation: OpenAI GPT (Chat) API

  • Making a chatting agent with user's pet information (user giving when joining the app).

2.2. Image Generation: Dreambooth Fast Stable Diffusion

3. Dataset (Demo Examples)

Dreambooth(kind of sd finetuning method) makes it possible to finetune the model with only a few images. Therefore, we do not use a large dataset. We only use data of 5 images for finetuning. Finetuning takes about 30 minutes. The finetuning outputs(ckpts) are as follows.

강아지 양파
1. Checkpoint: dhdbsrlw/pet-onion
2. Finetuning Images
(*src: Instagram @pom_onion)
onion_1 onion_2 onion_3 onion_4 onion_5
강아지 제로
1. Checkpoint: dhdbsrlw/pet-zero
2. Finetuning Images
(*src: Google)
zero_1 zero_2 zero_3 zero_4 zero_5
강아지 몽자
1. Checkpoint: dhdbsrlw/pet-monga
2. Finetuning Images
(*src: Instagram @mongja0408)
monga_1 monga_2 monga_3 monga_4 monga_5

4. Generation Output

Join Chat

More generation outputs can be accessed in here !



(Last Update: 2024/02/10)
Contact: [email protected]

About

2024 프로메테우스 해커톤 본선 진출작: AI 모델링 및 인프런스 서버 구축

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published