Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

paddle推理pipeline #381

Open
wants to merge 2 commits into
base: develop
Choose a base branch
from

Conversation

caizejun
Copy link

支持绕过fastdeploy,静态图直接用paddle_inference推理

Copy link

paddle-bot bot commented Jan 15, 2024

Thanks for your contribution!

print("PaddleInfer pipeline saved to", output_path)


# fd_pipe_cls = FastDeploySFastDeployStableDiffusionXLPipelineHousing

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这边是直接不用fd还是加个判断分别走呢?

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

我重写一份代码,用paddle_infer_pipeline导出

parser = argparse.ArgumentParser()
parser.add_argument(
"--model_dir",
default="runwayml/stable-diffusion-v1-5@fastdeploy",

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这个default是不是要改一下

@@ -444,6 +451,7 @@ def load_sub_model(
try:
# check if the module is in a subdirectory
if os.path.isdir(os.path.join(cached_folder, name)):
# import pdb; pdb.set_trace()

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

无用的debug 注释记得删一下

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants