请选择 进入手机版 | 继续访问电脑版

 找回密码
 立即注册
查看: 1141|回复: 0

[教程] 有国内开发者自己实现了 #Midjourney 官方的“根据图片解析 Prompt”功能,同时还支持

[复制链接]

3869

主题

36

回帖

4331

积分

管理员

积分
4331
发表于 2023-4-17 12:09:21 | 显示全部楼层 |阅读模式

有国内开发者自己实现了 #Midjourney 官方的“根据图片解析 Prompt”功能,同时还支持 Stable Diffusion,而且还能够基于描述进行扩展,以便二次图片生成。当然,它也可以根据那的原始描述生成绘画 Prompt 提示词。太强了!

prompt.png

下载地址:

https://github.com/soulteary/docker-prompt-generator

Docker 作图咒语生成器

使用模型来生成作图咒语的偷懒工具,支持 MidJourney、Stable Diffusion 等。

功能预览

如同 MidJourney 官方新推出的功能,工具支持一键从图片中解析出 Prompt 描述,并能够基于描述进行扩展,以便二次图片生成。

preview.jpg

工具支持直接使用中文进行原始 Prompt 描述,并能够将中文转换为模型生成效果更好的英文 Prompt 描述。

使用方法

在过去的几篇文章里,我提到过了我个人习惯和推荐的开发环境,基于 Docker 和 Nvidia 官方基础容器的深度学习环境,所以就不再赘述相关知识点,感兴趣可以自行翻阅,比如这篇《基于 Docker 的深度学习环境:入门篇》。相信老读者应该已经很熟悉啦。

当然,因为本文包含纯 CPU 也能玩的部分,你也可以参考几个月前的《在搭载 M1 及 M2 芯片 MacBook设备上玩 Stable Diffusion 模型》,来配置你的环境。

在准备好 Docker 环境的配置之后,我们就可以继续玩啦。

我们随便找一个合适的目录,使用 git clone 或者下载 Zip 压缩包的方式,先把“Docker Prompt Generator(Docker 作图咒语生成器)”项目的代码下载到本地。

git clone [https://github.com/soulteary/docker-prompt-generator.git](https://github.com/soulteary/docker-prompt-generator.git)
# or
curl -sL -o docker-prompt-generator.zip [https://github.com/soulteary/docker-prompt-generator/archive/refs/heads/main.zip](https://github.com/soulteary/docker-prompt-generator/archive/refs/heads/main.zip)

接着,进入项目目录,使用 Nvidia 原厂的 PyTorch Docker 基础镜像来完成基础环境的构建,相比于我们直接从 DockerHub 拉制作好的镜像,自行构建将能节约大量时间。

我们在项目目录中执行下面的命令,就能够完成应用模型应用的构建啦:

# 构建基础镜像
docker build -t soulteary/prompt-generator:base . -f docker/Dockerfile.base

# 构建 CPU 应用
docker build -t soulteary/prompt-generator:cpu . -f docker/Dockerfile.cpu

# 构建 GPU 应用
docker build -t soulteary/prompt-generator:gpu . -f docker/Dockerfile.gpu

然后,根据你的硬件环境,选择性执行下面的命令,就能够启动一个带有 Web UI 界面的模型应用啦。

# 运行 CPU 镜像
docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm -it -p 7860:7860 soulteary/prompt-generator:cpu

# 运行 GPU 镜像
docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm -it -p 7860:7860 soulteary/prompt-generator:gpu

我们在浏览器中输入运行容器的宿主机的 IP 地址,就能够开始使用工具啦。

相关项目

模型类项目:

数据集项目:

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

×
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Archiver|手机版|小黑屋|金房子

GMT+8, 2024-12-5 04:54 , Processed in 0.043429 second(s), 23 queries .

© 2023 金房子|AI发烧友社区

快速回复 返回顶部 返回列表