Instructions to use PleIAs/Baguettotron with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use PleIAs/Baguettotron with Transformers:

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

pipe = pipeline("text-generation", model="PleIAs/Baguettotron")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("PleIAs/Baguettotron")
model = AutoModelForCausalLM.from_pretrained("PleIAs/Baguettotron", device_map="auto")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use PleIAs/Baguettotron with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "PleIAs/Baguettotron"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "PleIAs/Baguettotron",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker

docker model run hf.co/PleIAs/Baguettotron

SGLang

How to use PleIAs/Baguettotron 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 "PleIAs/Baguettotron" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "PleIAs/Baguettotron",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

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 "PleIAs/Baguettotron" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "PleIAs/Baguettotron",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Docker Model Runner
How to use PleIAs/Baguettotron with Docker Model Runner:
```
docker model run hf.co/PleIAs/Baguettotron
```

pdelobelle commited on May 7

Commit

39792bd

verified ·

1 Parent(s): 2da2858

Delete figures

Browse files

Files changed (6) hide show

figures/baguettotron_structure.png +0 -3
figures/comparison_models.png +0 -3
figures/pleias.jpg +0 -0
figures/table_evaluation.png +0 -3
figures/training_baguettotron.png +0 -3
figures/training_efficiency.jpeg +0 -0

figures/baguettotron_structure.png DELETED Viewed

Git LFS Details

SHA256: ba862d564d77d7e8d87416298fa4b5e1af5dbfaeb4dc63b848f7b2f2197e7760
Pointer size: 131 Bytes
Size of remote file: 403 kB

figures/comparison_models.png DELETED Viewed

Git LFS Details

SHA256: d47c8cdb0bc5bca30738fa5059e20af373efd899ec924eb1b53d5a516bfac028
Pointer size: 131 Bytes
Size of remote file: 886 kB

figures/pleias.jpg DELETED Viewed

Binary file (45.4 kB)

figures/table_evaluation.png DELETED Viewed

Git LFS Details

SHA256: 1d64da90efb12166654bac389a6b348c3ffafe740a6cd392cba6c084c19a92b3
Pointer size: 131 Bytes
Size of remote file: 131 kB

figures/training_baguettotron.png DELETED Viewed

Git LFS Details

SHA256: 9abe7f56929daf01bf65d98629a76146decd9821c2faf2764d133099390d9b1c
Pointer size: 131 Bytes
Size of remote file: 329 kB

figures/training_efficiency.jpeg DELETED Viewed

Binary file (44.6 kB)