How to use mangsense/codet5-java-vulnerability-lora with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("Salesforce/codet5-small") model = PeftModel.from_pretrained(base_model, "mangsense/codet5-java-vulnerability-lora")
How to use mangsense/codet5-java-vulnerability-lora with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("mangsense/codet5-java-vulnerability-lora", device_map="auto")
https://huggingface.co/mangsense/codet5-java-vulnerability-lora/resolve/main/vocab.json
hf download hf://mangsense/codet5-java-vulnerability-lora/vocab.json
curl -L -o vocab.json https://huggingface.co/mangsense/codet5-java-vulnerability-lora/resolve/main/vocab.json
