agentscope
agentscope copied to clipboard
Add stable-diffusion model wrapper
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
And please pass the UT, thx!
I've made some updates to the PR:
- 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.
- 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.
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.
Closing for new version migration