[대규모 릴리즈] Supra2-IMG - 100M 파라미터의 초소형 텍스트-이미지 모델 - SOTA급 품질로 오픈 소스 공개!
[MASSIVE RELEASE] Supra2-IMG - a tiny 100M text-to-image model - SOTA quality and open release!
핵심 요약
100M 파라미터로 로컬 환경에서 빠르게 구동되는 초소형 텍스트-이미지 생성 모델이 공개되어 큰 관심을 받고 있습니다.
- 초소형 모델 — 100M 파라미터로 구성되어 CPU에서도 20초 내외로 이미지 생성 가능함
- 학습 효율성 — 단일 H100 GPU에서 10시간 미만으로 학습 완료됨
- 오픈 소스 — 허깅페이스를 통해 누구나 모델을 다운로드하고 로컬에서 실행 가능함
- 활용 가능성 — 엣지 디바이스에서의 빠른 이미지 생성 및 인페인팅 도구 등으로 활용 기대됨
다들 안녕!
마지막 SupraLabs 모델 이후로 꽤 오래 지났는데, 오늘 너희들을 위해 특별한 걸 하나 가져왔어: 바로 Supra2-IMG야.
이건 100M 파라미터짜리 DiT 텍스트-투-이미지 모델인데, Runpod에서 H100 한 장으로 10시간도 안 걸려서 밑바닥부터 싹 다 학습시켰어. 256x256 해상도에서 끝판왕급 퀄리티로 이미지를 뽑아낼 수 있다고.
샘플:
![r/LocalLLaMA - [MASSIVE RELEASE] Supra2-IMG - a tiny 100M text-to-image model - SOTA quality and open release!](https://preview.redd.it/massive-release-supra2-img-a-tiny-100m-text-to-image-model-v0-9paalvbs4wqh1.png?width=620&format=png&auto=webp&s=828b9e0fd967b82ddd94e9bf82e26da492185607)
이 샘플들 절대 잘 나온 것만 골라낸 거 아님! 샘플링 조건은 seed 0, steps 50, cfg 3.0으로 전부 다 똑같이 돌렸어.
혹시 프롬프트 궁금한 사람 있으면 알려줄 테니까 언제든 물어봐!
자기 컴퓨터에서도 로컬로 돌릴 수 있는데, CPU로 돌리면 20초 정도 걸리고 GPU로는 2초면 끝남 (🤩):
일단 이거부터 실행해:
Create project directory
mkdir Supra2-IMG
cd Supra2-IMG
Download the inference script
wget https://huggingface.co/SupraLabs/Supra2-IMG/resolve/main/inference.py
그다음엔 아래 명령어로 이미지 생성하면 돼:
python inference.py --prompt "a sea jellyfish floating in the pitch-black ocean depths" --seed 0 --cfg 3.0 --steps 50 --n 1 --out jellyfish.png
다들 재밌게 써봐 🤗 🔥
HF 모델 링크: https://huggingface.co/SupraLabs/Supra2-IMG
좋았으면 HF에서 좋아요랑 팔로우 좀 눌러줘 🤗 ❤️
피드백은 언제나 환영이야! 궁금한 거 있으면 뭐든 물어봐!
