我正在参与「兔了个兔」构思投稿大赛，详情请看：「兔了个兔」构思投稿大赛

前语

这不兔年到了，薅薅兔毛；

继帮小兔子完成了 作一首诗词，写一篇文章 之后，小兔子又有新的主意了；

小兔子说，新年到了，她想要一张自画像，可是她自己又画不来，拜托博主用 GPT 帮她画一个；

人美心善的博主当然是怅然答应了；

练习

预备一批练习集和一批验证集作为预练习的数据集，对两个数据集的图文对进行相似性打分，挑选 score 大于0.2的作为终究的预练习数据集；

初始化 text encoder，先只练习 text encoder，以便保存原始模型的生成才能且完成中文概念的对齐；

终究在一百万的图文对上练习了一个 epoch；

使用

1、导入对应的包：

from diffusers import StableDiffusionPipeline

TIPS

要注意 transform 和 torch 的版别，由于新旧版别有所改动，会出现版别不兼容的情况，要么便是修改源码，省力点的就直接将这两个包更新到最新；

博主这儿的版别更新变动如下：

  • transform : 4.17.0 -> 4.25.1
  • torch : 1.6.0 -> 1.13.0

2、初始化：

pipe = StableDiffusionPipeline.from_pretrained(check_point).to("cuda")

3、得到成果：

prompt = '兔子,新年,油画'
img = pipe(prompt, ...).images[0]
【AI】用 GPT 帮小兔子画肖像
【AI】用 GPT 帮小兔子画肖像
【AI】用 GPT 帮小兔子画肖像

解决问题

cannot import name 'EntryNotFoundError' from 'huggingface_hub.utils'

这个报错可以从字面进行了解，便是说没有 utils 中没有 EntryNotFoundError，直接进入源码进行查看：

...
from huggingface_hub.utils import EntryNotFoundError, RepositoryNotFoundError, RevisionNotFoundError
...
except RepositoryNotFoundError:
    raise EnvironmentError(
        f"{pretrained_model_name_or_path} is not a local folder and is not a valid model identifier"
        " listed on 'https://huggingface.co/models'\nIf this is a private repository, make sure to pass a"
        " token having permission to this repo with `use_auth_token` or log in with `huggingface-cli"
        " login`."
    )
except RevisionNotFoundError:
    raise EnvironmentError(
        f"{revision} is not a valid git identifier (branch name, tag name or commit id) that exists for"
        " this model name. Check the model page at"
        f" 'https://huggingface.co/{pretrained_model_name_or_path}' for available revisions."
    )
except EntryNotFoundError:
    raise EnvironmentError(
        f"{pretrained_model_name_or_path} does not appear to have a file named {cls.config_name}."
    )

发现只用在了 抛出反常 这块代码里，因而博主一开始就直接将这一段注释掉了，成果还有下一个报错；

cannot import name 'ModelOutput' from 'transformers.utils'

这个报错就属所以新旧版别的变动了，直接查看官方文档；

博主原先的 transformers 版别是 4.17.0，ModelOutput 是在 transformers.file_utils 里边的，如下图所示：

【AI】用 GPT 帮小兔子画肖像

而在最新的 transformers 版别 4.25.1 中，ModelOutput 是在 transformers.utils 里的，因而才引起的这个报错；

【AI】用 GPT 帮小兔子画肖像

当然可能在前面几个版别就已经改动了，这儿博主只看了最新版别的官方文档；

module 'torch.nn' has no attribute 'SiLU'

这个也是由于版别问题形成的；

博主之前的 torch 版别是 1.6.0，可以看到官方文档 torch.nn 中并没有 SiLU：

【AI】用 GPT 帮小兔子画肖像

可是下个版别 1.7.0 就有了，不过博主最终仍是更新到了 1.13.0 版别：

【AI】用 GPT 帮小兔子画肖像

后记

以上便是【AI】用 GPT 帮小兔子画肖像的全部内容了。

希望本篇博文对我们有所协助！

系列文章：

  • 【AI】用 GPT 帮小兔子作首诗
  • 【AI】用 GPT 帮小兔子写文章
  • 【AI】用 GPT 帮小兔子画肖像

上篇精讲：【AI】用 GPT 帮小兔子写文章

我是，期待你的关注；

创造不易，请多多支撑；

系列专栏：AI NLP