https://github.com/bitaffinity/ComfyUI_HF_Inference
ComfyUI nodes for Hugging Face hosted inference endpoints
https://github.com/bitaffinity/ComfyUI_HF_Inference
comfyui huggingface
Last synced: 6 months ago
JSON representation
ComfyUI nodes for Hugging Face hosted inference endpoints
- Host: GitHub
- URL: https://github.com/bitaffinity/ComfyUI_HF_Inference
- Owner: bitaffinity
- License: apache-2.0
- Created: 2024-05-30T22:22:36.000Z (about 1 year ago)
- Default Branch: main
- Last Pushed: 2024-06-10T22:46:12.000Z (12 months ago)
- Last Synced: 2024-06-11T01:32:40.598Z (12 months ago)
- Topics: comfyui, huggingface
- Language: Python
- Homepage:
- Size: 13.7 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
- awesome-comfyui - **ComfyUI_HF_Inference**
README
# Hugging Face hosted inference nodes for ComfyUI
Unofficial ComfyUI nodes for Hugging Face's inference API
Visit [the official docs](https://huggingface.co/docs/api-inference/detailed_parameters) for an overview of how the HF inference endpoints work
Find models by task on the [official website](https://huggingface.co/tasks)
## Installation
### Clone and install dependencies
```
git clone https://github.com/bitaffinity/ComfyUI_HF_Inference custom_nodes/ComfyUI_HF_Inference
cd custom_nodes/ComfyUI_HF_Inference
pip install -r requirements.txt
```
Export HF_AUTH_TOKEN with one of your [Hugging Face tokens](https://huggingface.co/settings/tokens)
### Run ComfyUI
`HF_AUTH_TOKEN=hf_1111111111111111111111111111111111 python main.py`
## Nodes
> [!WARNING]
> Inference API (serverless) requires a model 10GB or below and fails for random reasons on different models.
### Text
* Feature Extraction
- [facebook/bart-base](https://huggingface.co/facebook/bart-base)
* Question Answering
- [deepset/roberta-base-squad2](https://huggingface.co/deepset/roberta-base-squad2)
* Translation
- [google-t5/t5-base](https://huggingface.co/google-t5/t5-base)
* Generation
- [HuggingFaceH4/zephyr-7b-beta](https://huggingface.co/HuggingFaceH4/zephyr-7b-beta)
### Image
* Classification
- [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224)
* Object Detection
- [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50)
* Segmentation
- [facebook/detr-resnet-50-panoptic](https://huggingface.co/facebook/detr-resnet-50-panoptic)
* TextToImage
- [sd-community/sdxl-flash](https://huggingface.co/sd-community/sdxl-flash)