跳到主内容
@wquguru
精选88r/LocalLLaMA(Reddit)模型发布/更新

SupraLabs发布1亿参数文生图模型Supra2-IMG

[MASSIVE RELEASE] Supra2-IMG - a tiny 100M text-to-image model - SOTA quality and open release!

原文
发到 X
推荐理由

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:

首先,运行:

代码 · 1
# Create project directory mkdir Supra2-IMG cd Supra2-IMG # Download the inference script wget https://huggingface.co/SupraLabs/Supra2-IMG/resolve/main/inference.py
代码 · 1
# Create project directory mkdir Supra2-IMG cd Supra2-IMG # Download the inference script wget https://huggingface.co/SupraLabs/Supra2-IMG/resolve/main/inference.py

Then, you can generate images by running:

然后,你可以通过运行以下命令来生成图像:

代码 · 1
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
代码 · 1
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

Have 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!

我们非常欢迎每一位的反馈,朋友们!欢迎随时提出任何问题!

更进一步:量化金融体系

看懂新闻只是起点——沿量化金融路径,把它变成能交付的工程能力

进入量化体系 →

相似阅读

关联信息,但可能不是同一事件