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:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Azirqui/codet5-python-doc-generator-v2", dtype="auto")
12b88f3 f2ff67e 12b88f3
1
2
3
4
version https://git-lfs.github.com/spec/v1 oid sha256:a494431f5ab1ddd5148fbe0b7b82ef20622cf0ac61fe3c50e6b25257b33d1244 size 2369320
