Skip to content

Commit

Permalink
fix stablediffusion_model.py
Browse files Browse the repository at this point in the history
  • Loading branch information
cmgzn committed Sep 20, 2024
1 parent a54a453 commit 9f29e8e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/agentscope/models/stablediffusion_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import base64
import json
import time
from typing import Any, Optional, Union, List, Sequence
from typing import Any, Optional, Union, Sequence

import requests
from loguru import logger
Expand Down Expand Up @@ -231,7 +231,7 @@ def __call__(
# step3: parse the response
return self._parse_response(response)

def format(self, *args: Union[Msg, Sequence[Msg]]) -> List[dict] | str:
def format(self, *args: Union[Msg, Sequence[Msg]]) -> str:
# This is a temporary implementation to focus on the prompt
# on single-turn image generation by preserving only the system prompt
# and the last user message. This logic might change in the future
Expand Down

0 comments on commit 9f29e8e

Please sign in to comment.