David Chatel
ffb9c8f3a1
- Introduced DavchaLLMExtension class with node list retrieval. - Added DavchaLLMLoader, DavchaLLM, and DavchaPromptEnricher classes for model handling. - Implemented fuzzy matching for chat handler extraction in utils. - Included requirements for rapidfuzz library.
13 lines
380 B
Python
13 lines
380 B
Python
from comfy_api.latest import io, ComfyExtension
|
|
from .nodes import *
|
|
|
|
class DavchaLLMExtension(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
DavchaLLMLoader,
|
|
DavchaLLM,
|
|
DavchaPromptEnricher,
|
|
]
|
|
|
|
async def comfy_entrypoint() -> DavchaLLMExtension:
|
|
return DavchaLLMExtension() |