ComfyUI-HF

ComfyUI-HF
★ 20

模型加载Hugging FaceGPT-2ComfyUI 节点
在 ComfyUI 中访问 Hugging Face 仓库与模型的节点，支持从仓库加载模型并通过 GPT2Node 使用 GPT-2 生成提示，简化模型导入与推理。
💡 在 ComfyUI 中快速导入 Hugging Face 模型并生成文本提示。
🍴 6 Forks💻 Python🔄 2024-05-27
📦
网盘下载
复制链接后前往夸克网盘下载
https://pan.quark.cn/s/8f9eee5e2cdb
📦 requirements.txt
transformers==4.26.1
📄 README

ComfyUI-HF

ComfyUI Node to work with Hugging Face repositories

GPT2Node

Loads the given gp2 model to generate prompt

Models

  • [Gustavosta/MagicPrompt-Stable-Diffusion] (https://huggingface.co/Gustavosta/MagicPrompt-Stable-Diffusion)
  • daspartho/prompt-extend (~500mb)
  • succinctly/text2image-prompt-generator (~600mb) – Trained on Midjourney prompts
  • microsoft/Promptist (~500mb) – Read the paper here
  • AUTOMATIC/promptgen-lexart (~300mb) – Finetuned using 134,819 prompts from lexica.art
  • AUTOMATIC/promptgen-majinai-safe (~300mb) – 1,654 prompts from majinai.art
  • AUTOMATIC/promptgen-majinai-unsafe (~300mb) – 825 prompts from majinai.art (NSFW)
  • Gustavosta/MagicPrompt-Dalle
  • kmewhort/stable-diffusion-prompt-bolster (~500mb),
  • Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator (~500mb),
  • Ar4ikov/gpt2-medium-650k-stable-diffusion-prompt-generator (~1.4gb),
  • crumb/bloom-560m-RLHF-SD2-prompter-aesthetic (~1.1gb),
  • Meli/GPT2-Prompt (~500mb),
  • DrishtiSharma/StableDiffusion-Prompt-Generator-GPT-Neo-125M (~550mb)