cs-552-2026-catma
/

safety_model

Text Generation

Generated from Trainer

text-generation-inference

Model card Files Files and versions

Instructions to use cs-552-2026-catma/safety_model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use cs-552-2026-catma/safety_model with Transformers:

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

pipe = pipeline("text-generation", model="cs-552-2026-catma/safety_model")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("cs-552-2026-catma/safety_model")
model = AutoModelForCausalLM.from_pretrained("cs-552-2026-catma/safety_model", 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

How to use cs-552-2026-catma/safety_model with vLLM:

Install from pip and serve model

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

Use Docker

docker model run hf.co/cs-552-2026-catma/safety_model

How to use cs-552-2026-catma/safety_model 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 "cs-552-2026-catma/safety_model" \
    --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": "cs-552-2026-catma/safety_model",
		"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 "cs-552-2026-catma/safety_model" \
        --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": "cs-552-2026-catma/safety_model",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Docker Model Runner
How to use cs-552-2026-catma/safety_model with Docker Model Runner:
```
docker model run hf.co/cs-552-2026-catma/safety_model
```

Commit History

Automated MNLP evaluation report (2026-06-11) (#14)

c81e874

zechen-nlp commited on Jun 11

Automated MNLP evaluation report (2026-06-10) (#13)

d6a8b7c

zechen-nlp commited on Jun 10

Upload folder using huggingface_hub

e884f81
verified

ayarrafi commited on Jun 10

Delete training_args.bin with huggingface_hub

46dd178
verified

ayarrafi commited on Jun 10

Delete folder phase1 with huggingface_hub

18bab3d
verified

ayarrafi commited on Jun 10

Delete folder checkpoint-667 with huggingface_hub

8795ac5
verified

ayarrafi commited on Jun 10

Delete folder checkpoint-5498 with huggingface_hub

2353732
verified

ayarrafi commited on Jun 10

Delete folder checkpoint-500 with huggingface_hub

330f297
verified

ayarrafi commited on Jun 10

Delete folder checkpoint-4373 with huggingface_hub

e56e025
verified

ayarrafi commited on Jun 10

Delete folder checkpoint-319 with huggingface_hub

1510e2a
verified

ayarrafi commited on Jun 10

Delete folder checkpoint-250 with huggingface_hub

9351c18
verified

ayarrafi commited on Jun 10

Upload folder using huggingface_hub

730931a
verified

ayarrafi commited on Jun 10

Upload Qwen3ForCausalLM

87d5416
verified

ayarrafi commited on Jun 9

Automated MNLP evaluation report (2026-06-09) (#12)

f295cb2

zechen-nlp commited on Jun 9

Automated MNLP evaluation report (2026-06-08) (#11)

d649205

zechen-nlp commited on Jun 8

Upload tokenizer

ab6897c
verified

ayarrafi commited on Jun 8

Upload Qwen3ForCausalLM

6d74542
verified

ayarrafi commited on Jun 8

Automated MNLP evaluation report (2026-06-06) (#10)

9f69b48

zechen-nlp commited on Jun 6

Automated MNLP evaluation report (2026-06-05) (#9)

d4ed0be

zechen-nlp commited on Jun 5

Upload generation_config.json with huggingface_hub

b48e636
verified

TuanNguyen2003 commited on Jun 4

Upload folder using huggingface_hub

35705ff
verified

TuanNguyen2003 commited on Jun 4

Automated MNLP evaluation report (2026-06-03) (#8)

5abcd92

zechen-nlp commited on Jun 3

Upload folder using huggingface_hub

a4b7942
verified

TuanNguyen2003 commited on Jun 3

Automated MNLP evaluation report (2026-06-02) (#7)

f258d3b

zechen-nlp commited on Jun 3

Upload folder using huggingface_hub

26c8c74
verified

TuanNguyen2003 commited on Jun 1

Automated MNLP evaluation report (2026-06-01) (#6)

ab28c43

zechen-nlp commited on Jun 1

Upload folder using huggingface_hub

dd833f0
verified

TuanNguyen2003 commited on May 29

Automated MNLP evaluation report (2026-05-29) (#5)

ec1219d

zechen-nlp commited on May 29

Automated MNLP evaluation report (2026-05-28) (#4)

4cd24f9

zechen-nlp commited on May 28

Upload folder using huggingface_hub

684f403
verified

ayarrafi commited on May 27

Automated MNLP evaluation report (2026-05-27) (#3)

fce9420

zechen-nlp commited on May 27

Upload folder using huggingface_hub

2ea6f77
verified

ayarrafi commited on May 26

Automated MNLP evaluation report (2026-05-26) (#2)

48069d1

zechen-nlp commited on May 26

Delete folder checkpoint-2594 with huggingface_hub

72a00c2
verified

mariem38 commited on May 25

Delete folder checkpoint-1200 with huggingface_hub

efac0f6
verified

mariem38 commited on May 25

Delete model.safetensors.index.json with huggingface_hub

963c662
verified

mariem38 commited on May 25

Delete model-00002-of-00002.safetensors with huggingface_hub

2fa7199
verified

mariem38 commited on May 25

Delete model-00001-of-00002.safetensors with huggingface_hub

127bfe6
verified

mariem38 commited on May 25

Upload folder using huggingface_hub

f924d42
verified

mariem38 commited on May 25

Automated MNLP evaluation report (2026-05-23) (#1)

cb6e663

zechen-nlp commited on May 23

Safety model v3 - thinking traces, 6 options, rotation augmentation

e7b7386
verified

mariem38 commited on May 21

Update safety model v2 - multi-option MCQ training

0949f84
verified

mariem38 commited on May 16

Upload folder using huggingface_hub

a8bd1b9
verified

TuanNguyen2003 commited on May 14

initial commit

7dd8b15
verified

cristiano-masutti commited on May 11