Tomas245 commited on
Commit
89fafa3
·
verified ·
1 Parent(s): 0a0af2c

fixed typos in README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -86,7 +86,7 @@ print(safe_label, categories)
86
  from transformers import AutoModelForCausalLM, AutoTokenizer
87
  import re
88
 
89
- model_name = "Qwen/Qwen3Guard-4B-Gen"
90
 
91
  # load the tokenizer and the model
92
  tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -135,7 +135,7 @@ print(content)
135
  # Refusal: Yes
136
  # '''
137
  safe_label, category_label, refusal_label = extract_label_categories_refusal(content)
138
- print(safe_label, categories, refusal_label)
139
  ```
140
 
141
  ## Deployment with SGLang and vLLM
 
86
  from transformers import AutoModelForCausalLM, AutoTokenizer
87
  import re
88
 
89
+ model_name = "Qwen/Qwen3Guard-Gen-0.6B"
90
 
91
  # load the tokenizer and the model
92
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
135
  # Refusal: Yes
136
  # '''
137
  safe_label, category_label, refusal_label = extract_label_categories_refusal(content)
138
+ print(safe_label, category_label, refusal_label)
139
  ```
140
 
141
  ## Deployment with SGLang and vLLM