prem-research
/

MiniGuard-v0.1

Text Generation

text-generation-inference

Model card Files Files and versions

ojus1 commited on 8 days ago

Commit

ba55737

·

verified ·

1 Parent(s): e9ce077

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -116,7 +116,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 from jinja2 import Template
-model_name = "nvidia/Llama-3.1-Nemotron-Safety-Guard-8B-v3"
 # Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 from jinja2 import Template
+model_name = "prem-research/MiniGuard-v0.1"
 # Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_name)