dimaboyko / huggingface-vscode-endpoint-server

starcoder server for huggingface-vscdoe custom endpoint

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Hugging Face VSCode Endpoint Server

starcoder server for huggingface-vscdoe custom endpoint.

Can't handle distributed inference very well yet.

Usage

pip install -r requirements.txt
python main.py

Fill http://localhost:8000/api/generate/ into Hugging Face Code > Model ID or Endpoint in VSCode.

API

curl -X POST http://localhost:8000/api/generate/ -d '{"inputs": "", "parameters": {"max_new_tokens": 64}}'
# response = {"generated_text": ""}

About

starcoder server for huggingface-vscdoe custom endpoint

License:Apache License 2.0


Languages

Language:Python 100.0%