agentscope icon indicating copy to clipboard operation
agentscope copied to clipboard

Add stable-diffusion model wrapper

Open cmgzn opened this issue 1 year ago • 3 comments


name: Pull Request about: Create a pull request

Description

add stable-diffusion model wrapper Integrate https://github.com/AUTOMATIC1111/stable-diffusion-webui

Checklist

Please check the following items before code is ready to be reviewed.

  • [x] Code has passed all tests
  • [x] Docstrings have been added/updated in Google Style
  • [ ] Documentation has been updated
  • [x] Code is ready for review

cmgzn avatar Sep 11 '24 03:09 cmgzn

And please pass the UT, thx!

rayrayraykk avatar Sep 18 '24 12:09 rayrayraykk

I've made some updates to the PR:

  1. Replaced the custom request with the third-party package webuiapi. Let me know if this is suitable for the project or if you prefer a different approach.
  2. I've wrapped the SD model as a callable agent service (similar to 'dashscope_services'), as I realized that directly conversing with the text2img model was not a good idea. But the dialogue agent for direct conversation with the SD model is still available and can be modified based on your feedback.

cmgzn avatar Sep 27 '24 10:09 cmgzn

Could you please further provide a minimal hardware requirement for running this demo on README? And the other part looks good to me. @DavdGao Please have a look at this PR.

rayrayraykk avatar Sep 29 '24 06:09 rayrayraykk

Closing for new version migration

DavdGao avatar Aug 13 '25 14:08 DavdGao