KHongJae/Chatting_Based_Emoji_Generation_Model

T-Academy ASAC 6기 DL 프로젝트에서 사용한 이모티콘 생성 모델입니다.
기존 묘사형 프롬프트에서만 작동 하던 Stable Diffusion을 대화형 프롬프트에서 작동할 수 있도록 시도한 모델입니다.

Intended uses & limitations

• prompt : '파란 귀를 가진 고양이, 이모티콘, 단색 배경, 디저트 먹는 중!'

How to use

pipeline = DiffusionPipeline.from_pretrained(
  "KHongJae/Chatting_Based_Emoji_Generation_Model",
  torch_dtype=torch.float16
).to("cuda")
pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
prompt = "Create your own prompt"
negative_prompt = "Create your own negative prompt"

pipeline(
  prompt=prompt,
  negative_prompt=negative_prompt,
  width=200,
  height=200,
  num_inference_steps=50,
  num_images_per_prompt=1
).images[0]

Training details

• 카카오톡 이모티콘 데이터
• ChatGPT4