Instructions to use mangsense/codet5-java-vulnerability-lora with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
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") - Transformers
How to use mangsense/codet5-java-vulnerability-lora with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("mangsense/codet5-java-vulnerability-lora", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Upload requirements.txt
Browse files- requirements.txt +3 -0
requirements.txt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
peft
|
| 2 |
+
transformers
|
| 3 |
+
torch
|