fix: fixed stability adapter (#48) (#49)

epam · Dec 21, 2023 · a6bdd06 · a6bdd06
1 parent bff58ff
commit a6bdd06
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/aidial_adapter_bedrock/llm/model/stability.py b/aidial_adapter_bedrock/llm/model/stability.py
@@ -96,19 +96,17 @@ class StabilityAdapter(ChatModel):
     client: Bedrock
     storage: Optional[FileStorage]
 
-    def __init__(
-        self, client: Bedrock, model: str, storage: Optional[FileStorage]
-    ):
-        super().__init__(model=model, tools_emulator=default_tools_emulator)
-        self.client = client
-        self.storage = storage
-
     @classmethod
     def create(cls, client: Bedrock, model: str, headers: Mapping[str, str]):
         storage: Optional[FileStorage] = create_file_storage(
             "images/stable-diffusion", headers
         )
-        return cls(client, model, storage)
+        return cls(
+            client=client,
+            model=model,
+            storage=storage,
+            tools_emulator=default_tools_emulator,
+        )
 
     def _prepare_prompt(
         self, messages: List[BaseMessage], max_prompt_tokens: Optional[int]