Instructions to use GItaf/bert2bert-with-cross-attn-decoder with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use GItaf/bert2bert-with-cross-attn-decoder with Transformers:

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

pipe = pipeline("text-generation", model="GItaf/bert2bert-with-cross-attn-decoder")

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("GItaf/bert2bert-with-cross-attn-decoder")
model = AutoModelForCausalLM.from_pretrained("GItaf/bert2bert-with-cross-attn-decoder", device_map="auto")

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use GItaf/bert2bert-with-cross-attn-decoder with vLLM:

Install from pip and serve model

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

Use Docker

docker model run hf.co/GItaf/bert2bert-with-cross-attn-decoder

SGLang

How to use GItaf/bert2bert-with-cross-attn-decoder 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 "GItaf/bert2bert-with-cross-attn-decoder" \
    --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": "GItaf/bert2bert-with-cross-attn-decoder",
		"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 "GItaf/bert2bert-with-cross-attn-decoder" \
        --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": "GItaf/bert2bert-with-cross-attn-decoder",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use GItaf/bert2bert-with-cross-attn-decoder with Docker Model Runner:
```
docker model run hf.co/GItaf/bert2bert-with-cross-attn-decoder
```

bert2bert-with-cross-attn-decoder

Commit History

Dec 10, 2024

06:06
Adding `safetensors` variant of this model⁠

SFconvertbot

Verified signature
14f2a8a

Sep 7, 2022

11:58
Upload pytorch_model.bin

GItaf

5939234
11:57
add

BrandonC8310

423d12e
11:36
delete

BrandonC8310

70c46f7
10:26
Upload pytorch_model.bin

GItaf

db48488
10:25
Upload training_args.bin

GItaf

3ba9372
10:25
Upload tokenizer.json

GItaf

82ea985
10:25
Upload tokenizer_config.json

GItaf

ee27628
10:24
Upload special_tokens_map.json

GItaf

a872719
10:24
Upload config.json

GItaf

5360a55

Sep 5, 2022

11:00
update model card README.md

GItaf

6260a4b
08:54
Training in progress, epoch 15

GItaf

1d197ba
08:29
Training in progress, epoch 14

GItaf

f27a3cc
08:10
Training in progress, epoch 13

GItaf

a314787
07:57
Training in progress, epoch 12

GItaf

f7a81c4
07:43
Training in progress, epoch 11

GItaf

31def85
05:13
Training in progress, epoch 10

GItaf

51d8b84
05:00
Training in progress, epoch 9

GItaf

6f97c02
04:46
Training in progress, epoch 8

GItaf

276c0ea
04:32
Training in progress, epoch 7

GItaf

f807317
04:19
Training in progress, epoch 6

GItaf

60440bb

Sep 4, 2022

19:15
Training in progress, epoch 5

GItaf

113fcac
19:01
Training in progress, epoch 4

GItaf

dea8240
18:48
Training in progress, epoch 3

GItaf

30abfa5
18:34
Training in progress, epoch 2

GItaf

9e44caf
18:21
Training in progress, epoch 1

GItaf

cacfbfc
18:08
initial commit

GItaf

3364a40

Commit History

Dec 10, 2024

Adding `safetensors` variant of this model⁠

Sep 7, 2022

Upload pytorch_model.bin

add

delete

Upload pytorch_model.bin

Upload training_args.bin

Upload tokenizer.json

Upload tokenizer_config.json

Upload special_tokens_map.json

Upload config.json

Sep 5, 2022

update model card README.md

Training in progress, epoch 15

Training in progress, epoch 14

Training in progress, epoch 13

Training in progress, epoch 12

Training in progress, epoch 11

Training in progress, epoch 10

Training in progress, epoch 9

Training in progress, epoch 8

Training in progress, epoch 7

Training in progress, epoch 6

Sep 4, 2022

Training in progress, epoch 5

Training in progress, epoch 4

Training in progress, epoch 3

Training in progress, epoch 2

Training in progress, epoch 1

initial commit