SupraLabs发布1亿参数文生图模型Supra2-IMG
[MASSIVE RELEASE] Supra2-IMG - a tiny 100M text-to-image model - SOTA quality and open release!
1亿参数的文生图模型能跑出SOTA画质且训练极快,对边缘端部署极具参考价值,建议关注其技术细节。
Hey everyone!
大家好!
It has been quite a while since the last SupraLabs model - but today we've something special for y'all: Supra2-IMG
距离上一个 SupraLabs 模型已经过了相当长的一段时间——但今天我们要为大家带来特别的内容:Supra2-IMG
It's a 100M parameter DiT text-to-image model trained entirely from scratch in under 10 hours on a single H100 on Runpod. It can generate state-of-the-art quality images in 256x256 pixels resolution.
这是一个拥有 1 亿参数的 DiT 文本到图像模型,完全从零开始训练,在 Runpod 的单张 H100 GPU 上耗时不到 10 小时。它能够在 256x256 像素分辨率下生成最先进的质量图像。
Samples:
示例:
https://preview.redd.it/9paalvbs4wqh1.png?width=620&format=png&auto=webp&s=ddc7876d187e4c6d349149c75baa210b0219a041
https://preview.redd.it/9paalvbs4wqh1.png?width=620&format=png&auto=webp&s=ddc7876d187e4c6d349149c75baa210b0219a041
These samples are NOT cherry-picked! Sampling: seed 0, steps 50, cfg 3.0; same settings for every image.
这些样本并非精心挑选!采样参数:种子值 0,步数 50,cfg 3.0;每张图像均使用相同设置。
If someone here is interested in the prompts, I can give them to you! Feel free to ask!
如果有人对提示词感兴趣,我可以提供给你们!欢迎随时提问!
You can also use the model locally on your hardware (~20s for an image on CPU (🤩) and ~2s for an image on GPU):
你也可以在自己的硬件上本地运行该模型(CPU 上生成一张图像约需 20 秒(🤩),GPU 上约需 2 秒):
First, run:
首先,运行:
# Create project directory mkdir Supra2-IMG cd Supra2-IMG # Download the inference script wget https://huggingface.co/SupraLabs/Supra2-IMG/resolve/main/inference.py# Create project directory mkdir Supra2-IMG cd Supra2-IMG # Download the inference script wget https://huggingface.co/SupraLabs/Supra2-IMG/resolve/main/inference.pyThen, you can generate images by running:
然后,你可以通过运行以下命令来生成图像:
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.pngpython inference.py --prompt "a sea jellyfish floating in the pitch-black ocean depths" --seed 0 --cfg 3.0 --steps 50 --n 1 --out jellyfish.pngHave fun 🤗 🔥
玩得开心 🤗 🔥
Link to the model on HF: https://huggingface.co/SupraLabs/Supra2-IMG
HF 上的模型链接:https://huggingface.co/SupraLabs/Supra2-IMG
Give us a like and a follow on HF if you want 🤗 ❤️
如果你愿意,请在 HF 上给我们点赞和关注 🤗 ❤️
EVERY feedback is welcome, guys! Feel free to ask any questions!
我们非常欢迎每一位的反馈,朋友们!欢迎随时提出任何问题!
更进一步:量化金融体系
看懂新闻只是起点——沿量化金融路径,把它变成能交付的工程能力