diff --git a/nodes/feeling_lucky.py b/nodes/feeling_lucky.py index 8d13b17..c7ce89d 100644 --- a/nodes/feeling_lucky.py +++ b/nodes/feeling_lucky.py @@ -30,9 +30,8 @@ def get_prompt(self, text: str, seed: int, autorefresh: str) -> tuple[str]: try: prompt = self._prompt_generator.generate(text, 1)[0] - print(prompt) - return (prompt,) + return (str(prompt),) except Exception as e: logger.exception(e) return ("",) diff --git a/nodes/jinja.py b/nodes/jinja.py index f0e8bc1..7cdc96e 100644 --- a/nodes/jinja.py +++ b/nodes/jinja.py @@ -8,4 +8,4 @@ def generate_prompt(self, text): prompt_generator = JinjaGenerator() all_prompts = prompt_generator.generate(text, 1) or [""] - return all_prompts[0] + return str(all_prompts[0]) diff --git a/nodes/magicprompt.py b/nodes/magicprompt.py index c4b4f57..11eada2 100644 --- a/nodes/magicprompt.py +++ b/nodes/magicprompt.py @@ -33,9 +33,8 @@ def get_prompt(self, text: str, seed: int, autorefresh: str) -> tuple[str]: try: prompt = self._prompt_generator.generate(text, 1)[0] - print(prompt) - return (prompt,) + return (str(prompt),) except Exception as e: logger.exception(e) return ("",) diff --git a/nodes/sampler.py b/nodes/sampler.py index cf8d6cb..5b18ce0 100644 --- a/nodes/sampler.py +++ b/nodes/sampler.py @@ -105,7 +105,7 @@ def get_prompt(self, text: str, seed: int, autorefresh: str) -> tuple[str]: new_prompt = self._get_next_prompt(self._prompts, self._current_prompt) print(f"New prompt: {new_prompt}") - return (new_prompt,) + return (str(new_prompt),) @abstractproperty def context(self) -> SamplingContext: