Hugging Face's logo

ToxicityPrompts
/
PolyGuard-Qwen-Smol

Text2Text Generation
Safetensors
qwen2
Model card Files Community
1
felfri commited on
Commit
7547cb3
·
verified ·
1 Parent(s): 7f1ad38

Update README.md

Browse files

fix import statement

Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -30,6 +30,8 @@ Abstract: Truly multilingual safety moderation efforts for Large Language Models
30
 
31
  ```python
32
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
33
  model_id = "ToxicityPrompts/PolyGuard-Qwen-Smol"
34
  tokenizer = AutoTokenizer.from_pretrained(model_id)
35
  model = AutoModelForCausalLM.from_pretrained(model_id)
 
30
 
31
  ```python
32
  from transformers import AutoTokenizer, AutoModelForCausalLM
33
+ import torch
34
+
35
  model_id = "ToxicityPrompts/PolyGuard-Qwen-Smol"
36
  tokenizer = AutoTokenizer.from_pretrained(model_id)
37
  model = AutoModelForCausalLM.from_pretrained(model_id)