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