include StableDiffusionImageSynthesisWrapper in models init

modelscope · Jan 9, 2025 · 612b1b6 · 612b1b6
1 parent edd5d5e
commit 612b1b6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/agentscope/models/__init__.py b/src/agentscope/models/__init__.py
@@ -62,6 +62,7 @@
     "LiteLLMChatWrapper",
     "YiChatWrapper",
     "AnthropicChatWrapper",
+    "StableDiffusionImageSynthesisWrapper",
 ]
 
 __all__ = [