Instructions to use Voicelab/vlt5-base-keywords with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use Voicelab/vlt5-base-keywords with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="Voicelab/vlt5-base-keywords")

# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("Voicelab/vlt5-base-keywords")
model = AutoModelForSeq2SeqLM.from_pretrained("Voicelab/vlt5-base-keywords", device_map="auto")

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use Voicelab/vlt5-base-keywords with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "Voicelab/vlt5-base-keywords"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "Voicelab/vlt5-base-keywords",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker

docker model run hf.co/Voicelab/vlt5-base-keywords

SGLang

How to use Voicelab/vlt5-base-keywords with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "Voicelab/vlt5-base-keywords" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "Voicelab/vlt5-base-keywords",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker images

docker run --gpus all \
    --shm-size 32g \
    -p 30000:30000 \
    -v ~/.cache/huggingface:/root/.cache/huggingface \
    --env "HF_TOKEN=<secret>" \
    --ipc=host \
    lmsysorg/sglang:latest \
    python3 -m sglang.launch_server \
        --model-path "Voicelab/vlt5-base-keywords" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "Voicelab/vlt5-base-keywords",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use Voicelab/vlt5-base-keywords with Docker Model Runner:
```
docker model run hf.co/Voicelab/vlt5-base-keywords
```

RyanZhuuuuu commited on Feb 7, 2024

Commit

8a91a9c

verified ·

1 Parent(s): 89a069b

Add Core ML conversion

Browse files

Core ML conversion, task=text2text-generation, precision=float32

Files changed (6) hide show

coreml/text2text-generation/decoder_float32_model.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
coreml/text2text-generation/decoder_float32_model.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
coreml/text2text-generation/decoder_float32_model.mlpackage/Manifest.json +18 -0
coreml/text2text-generation/encoder_float32_model.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
coreml/text2text-generation/encoder_float32_model.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
coreml/text2text-generation/encoder_float32_model.mlpackage/Manifest.json +18 -0

coreml/text2text-generation/decoder_float32_model.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:726dd2497bf5b1ba456256b3e67d18295f4c957474163fcede71fd0246d61923
+size 223123

coreml/text2text-generation/decoder_float32_model.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:a9652a7072ff2d454aedc1176455e04a5b54db4b5b8fee01c4931eccd1f21b45
+size 762389056

coreml/text2text-generation/decoder_float32_model.mlpackage/Manifest.json ADDED Viewed

	@@ -0,0 +1,18 @@

+{
+    "fileFormatVersion": "1.0.0",
+    "itemInfoEntries": {
+        "2D1D120D-5443-4EDA-8B19-93B89A0F251A": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
+        },
+        "33020D2E-D37F-4E3D-A7A8-FDCAB76FE18D": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Weights",
+            "name": "weights",
+            "path": "com.apple.CoreML/weights"
+        }
+    },
+    "rootModelIdentifier": "2D1D120D-5443-4EDA-8B19-93B89A0F251A"
+}

coreml/text2text-generation/encoder_float32_model.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:60f9622d77c4628682514838e13dd7b33b5d387a7b82adfbd053c2cdcd4b31cf
+size 123674

coreml/text2text-generation/encoder_float32_model.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:e359285b8c809fc6f0d165970dfb518b4065be5bf33efaf3ebf8f220ae08d0b4
+size 494367872

coreml/text2text-generation/encoder_float32_model.mlpackage/Manifest.json ADDED Viewed

	@@ -0,0 +1,18 @@

+{
+    "fileFormatVersion": "1.0.0",
+    "itemInfoEntries": {
+        "8B23E2F6-8938-44B9-AB89-169AF50B9FC4": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Specification",
+            "name": "model.mlmodel",
+            "path": "com.apple.CoreML/model.mlmodel"
+        },
+        "B39FF586-468A-4243-B489-15D29385554B": {
+            "author": "com.apple.CoreML",
+            "description": "CoreML Model Weights",
+            "name": "weights",
+            "path": "com.apple.CoreML/weights"
+        }
+    },
+    "rootModelIdentifier": "8B23E2F6-8938-44B9-AB89-169AF50B9FC4"
+}