PaddleHub:使用文心大模型ERNIE-ViLG进行高质量文图生成 - 飞桨AI Studio
我安装的时候在 opencv-python 卡住了,换了源就过去了:
-i https://pypi.tuna.tsinghua.edu/simple:
pip install --upgrade paddlehub -i https://pypi.tuna.tsinghua.edu/simple
pip install docarray
import paddlehub as hub
from docarray import DocumentArray, Document
ernie_vilg_module = hub.Module(name='ernie_vilg')
函数定义描述:
def generate_image(
text_prompts: str,
style: Optional[str] = "油画",
topk: Optional[int] = 2,
output_dir: Optional[str] = 'ernievilg_output')
参数描述:
- text_prompts(str): 输入的语句,描述想要生成的图像的内容。
- style(Optional[str]): 生成图像的风格,当前支持'油画','水彩','粉笔画','卡通','儿童画','蜡笔画'。
- topk(Optional[int]): 保存前多少张图,最多保存10张。
- output_dir(Optional[str]): 保存输出图像的目录,默认为"ernievilg_output"。
应用:
python
import paddlehub as hub
from docarray import DocumentArray, Document
ernie_vilg_module = hub.Module(name='ernie_vilg')
results = ernie_vilg_module.generate_image(text_prompts="早上的阳光懒懒的照在身上", style="油画", output_dir='/Users/manzhou/Downloads/paddleoutput')
应用2(提示异常):
import paddlehub as hub
from docarray import DocumentArray, Document
ernie_vilg_module = hub.Module(name='ernie_vilg')
results = ernie_vilg_module.generate_image(text_prompts="早上的阳光懒懒的照在身上", style="油画", topk=2, output_dir='/Users/manzhou/Downloads/paddleoutput')
展示下结果 :
更多推荐
mac OS飞桨环境搭建-玩PaddleHub大模型
发布评论