Duplicated from geoffmunn/Qwen3Guard-StarTrek-Classification-0.6B
How to use vllm-project/Qwen3Guard-StarTrek-Classification-0.6B with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("Qwen/Qwen3-0.6B") model = PeftModel.from_pretrained(base_model, "vllm-project/Qwen3Guard-StarTrek-Classification-0.6B")
How to use vllm-project/Qwen3Guard-StarTrek-Classification-0.6B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="vllm-project/Qwen3Guard-StarTrek-Classification-0.6B")
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("vllm-project/Qwen3Guard-StarTrek-Classification-0.6B", device_map="auto")
How to fix it?
