Add ERNIE-ViLG models into Pipelines #3512
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
wawltor
reviewed
Oct 20, 2022
|
|
||
| ### 3.2 一键体验文生图系统 | ||
|
|
||
| 在运行下面的命令之前，需要在官网申请`API Key`和 `Secret key`两个密钥，然后执行下面的命令。 |
|
|
||
| WebUI使用了[gradio前端](https://gradio.app/)，首先需要安装gradio，运行命令如下： | ||
| ``` | ||
| pip install gradio |
There was a problem hiding this comment.
后续看看gradio要不要放到requirments里面
There was a problem hiding this comment.
如果以后都是基于Gradio开发，我觉得可以
| from pipelines.nodes.base import BaseComponent | ||
|
|
||
|
|
||
| class ErnieVilGImageGenerator(BaseComponent): |
There was a problem hiding this comment.
这个咱们要不要区分一下 输入AK、SK这种的API形式，和普通的文生图的形式在命名上要不要区分开
后续咱们piplines需要部署模型形式的
同时建议Componet的命令和领域名字相关，ImageGenerator 也包括了GAN相关的图片生成方式
There was a problem hiding this comment.
已经换成ErnieTextToImageGenerator
| @@ -263,3 +264,34 @@ def run(self, | |||
| """ | |||
| output = self.pipeline.run(query=query, params=params, debug=debug) | |||
| return output | |||
|
|
|||
|
|
|||
| class ImageGenerationPipeline(BaseStandardPipeline): | |||
There was a problem hiding this comment.
同上，命名需要注意一下
这里需要讨论一下，后续所有的文生图的pipelines都要这个pipeline吗？
There was a problem hiding this comment.
已经修改成TextToImagePipeline
| @@ -81,6 +81,27 @@ def query(request: QueryRequest): | |||
| return result | |||
|
|
|||
|
|
|||
| @router.post("/query_images", | |||
There was a problem hiding this comment.
已经改成query_text_to_images
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
PR changes
Description