mirror of
https://github.com/wassname/peft.git
synced 2026-09-09 11:28:32 +08:00
@@ -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
|
||||
)
|
||||
|
||||
|
||||
@@ -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={},
|
||||
|
||||
Reference in New Issue
Block a user