mac OS飞桨环境搭建-玩PaddleHub大模型

PaddleHub：使用文心大模型ERNIE-ViLG进行高质量文图生成 - 飞桨AI Studio

我安装的时候在 opencv-python 卡住了，换了源就过去了：

-i https://pypi.tuna.tsinghua.edu.cn/simple：

pip install --upgrade paddlehub -i https://pypi.tuna.tsinghua.edu.cn/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')

展示下结果 ：