52 lines
1.7 KiB
Markdown
52 lines
1.7 KiB
Markdown
# Documentation
|
|
- Class name: FooocusPromptExpansion
|
|
- Category: prompt
|
|
- Output node: True
|
|
- Repo Ref: https://github.com/esheep/esheep_custom_nodes.git
|
|
|
|
该节点旨在根据输入文本和种子值扩展和生成新文本，种子值引入了随机性，旨在增强生成内容的多样性和创造性，同时通过种子值保持一定程度的可控性。
|
|
|
|
# Input types
|
|
## Required
|
|
- text
|
|
- 文本参数是节点文本扩展过程的源材料。它至关重要，因为它为新内容的生成提供了上下文和基础。
|
|
- Comfy dtype: STRING
|
|
- Python dtype: str
|
|
## Optional
|
|
- seed
|
|
- 种子参数为文本扩展过程引入随机性，允许从相同的输入文本生成多样化的输出。
|
|
- Comfy dtype: INT
|
|
- Python dtype: int
|
|
|
|
# Output types
|
|
- result
|
|
- 结果参数包含扩展后的文本，是节点文本扩展过程的输出。
|
|
- Comfy dtype: STRING
|
|
- Python dtype: List[str]
|
|
|
|
# Usage tips
|
|
- Infra type: CPU
|
|
|
|
# Source code
|
|
```
|
|
class FooocusPromptExpansion:
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {'required': {'text': ('STRING', {'multiline': True}), 'seed': ('INT', {'default': 0, 'min': 0, 'max': 18446744073709551615})}}
|
|
RETURN_TYPES = ('STRING',)
|
|
FUNCTION = 'main'
|
|
CATEGORY = 'prompt'
|
|
OUTPUT_IS_LIST = (True,)
|
|
OUTPUT_NODE = True
|
|
|
|
@staticmethod
|
|
def scale_seed(seed):
|
|
return seed % (2 ** 32 - 1)
|
|
|
|
def main(self, text, seed):
|
|
seed = FooocusPromptExpansion.scale_seed(seed)
|
|
res = do_expansion(text, seed)
|
|
logger.info(f'[FooocusPromptExpansion] (seed={seed}) expand |{text}| to |{res}|')
|
|
return {'ui': {'result': [res]}, 'result': ([res],)}
|
|
``` |