[대규모 공개] Supra2-IMG - 1억 개 매개변수의 소형 텍스트-이미지 모델 - SOTA 품질 및 오픈 공개!

여러분 안녕하세요! 지난 SupraLabs 모델 이후로 시간이 꽤 지났지만, 오늘 여러분을 위해 특별한 것을 가져왔습니다: Supra2-IMG입니다. 이 모델은 Runpod의 단일 H100 GPU에서 10시간도 채 걸리지 않아 완전히 처음부터 학습된 1억 개 매개변수의 DiT(Diffusion Transformer) 텍스트-이미지 모델입니다. 256x256 픽셀 해상도로 최첨단 품질의 이미지를 생성할 수 있습니다. 샘플: https://preview.redd.it/9paalvbs4wqh1.png?width=620&format=png&auto=webp&s=ddc7876d187e4c6d349149c75baa210b0219a041 이 샘플들은 임의로 고른 것이 아닙니다! 샘플링 설정: seed 0, steps 50, cfg 3.0; 모든 이미지에 동일한 설정이 적용되었습니다. 혹시 프롬프트가 궁금하신 분이 있다면 알려드릴 수 있습니다! 편하게 물어봐 주세요! 또한 이 모델을 로컬 하드웨어에서 사용하실 수도 있습니다 (~CPU 기준 이미지 생성 시간 약 20초 (🤩), GPU 기준 약 2초): 먼저 다음 명령어를 실행하세요: # 프로젝트 디렉토리 생성 mkdir Supra2-IMG cd Supra2-IMG # 추론 스크립트 다운로드 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 팔로우를 해주시면 감사하겠습니다 🤗 ❤️ 모든 피드백을 환영합니다, 여러분! 궁금한 점이 있으면 언제든지 질문해 주세요! submitted by /u/LH-Tech_AI [link] [comments]