diff --git a/README.md b/README.md index b91023a..792d467 100644 --- a/README.md +++ b/README.md @@ -34,11 +34,11 @@ Supported methods: ```python from transformers import AutoModelForSeq2SeqLM -from peft import get_peft_config, get_peft_model, LoRAConfig, TaskType +from peft import get_peft_config, get_peft_model, LoraConfig, TaskType model_name_or_path = "bigscience/mt0-large" tokenizer_name_or_path = "bigscience/mt0-large" -peft_config = LoRAConfig( +peft_config = LoraConfig( task_type=TaskType.SEQ_2_SEQ_LM, inference_mode=False, r=8, lora_alpha=32, lora_dropout=0.1 ) diff --git a/setup.py b/setup.py index b727d65..e1b34b6 100644 --- a/setup.py +++ b/setup.py @@ -22,7 +22,7 @@ extras["dev"] = extras["quality"] + extras["docs_specific"] setup( name="peft", - version="0.1.0.dev", + version="0.1.0.dev0", description="Parameter-Efficient Fine-Tuning (PEFT)", long_description=open("README.md", "r", encoding="utf-8").read(), long_description_content_type="text/markdown", @@ -30,7 +30,7 @@ setup( license="Apache", author="The HuggingFace team", author_email="sourab@huggingface.co", - url="https://github.com/huggingface/pets", + url="https://github.com/huggingface/peft", package_dir={"": "src"}, packages=find_packages("src"), entry_points={},