Update README.md

README.md

@@ -1,5 +1,12 @@
 ---
 license: apache-2.0
+datasets:
+- songff/UltraPrompt
+language:
+- en
+base_model:
+- meta-llama/Llama-3.2-3B-Instruct
+library_name: transformers
 ---
 
 # P-Aligner
@@ -8,14 +15,14 @@ license: apache-2.0
 ## Quick Start
 
 ```python
-import vllm
+from vllm import LLM, SamplingParams
 from transformers import AutoTokenizer
 
 raw_instruction = "What is the capital of France?"
-model_path = "songff/P-Aligner"
+model_path = "P-Aligner"
 
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
-model = vllm.LLM(
+model = LLM(
     model=model_path,
     gpu_memory_utilization=0.9,
     enable_prefix_caching=True,
@@ -24,6 +31,10 @@ model = vllm.LLM(
 
 outputs = model.generate(
     [raw_instruction],
+    sampling_params=SamplingParams(
+        temperature=0.0,
+        max_tokens=2048,
+    ),
 )
 better_instruction = tokenizer.parse_output(
     outputs[0].outputs[0].text,