Buckets:

|
download
raw
50 kB
# Models
[PeftModel](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel) is the base model class for specifying the base Transformer model and configuration to apply a PEFT method to. The base `PeftModel` contains methods for loading and saving models from the Hub.
## PeftModel[[peft.PeftModel]]
- **model** ([PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel)) -- The base transformer model used for Peft.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) -- The configuration of the Peft model.
- **adapter_name** (`str`, *optional*) -- The name of the adapter, defaults to `"default"`.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter weights
using float16 and bfloat16 to float32, as this is typically required for stable training, and only affect
select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the corresponding layer.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device. Useful to speed up the loading loading process.
> [!TIP] > Don't use `low_cpu_mem_usage=True` when creating a new PEFT adapter for training.
Base model encompassing various Peft methods.
**Attributes**:
- **base_model** (`torch.nn.Module`) -- The base transformer model used for Peft.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) -- The configuration of the Peft model.
- **modules_to_save** (`list` of `str`) -- The list of sub-module names to save when
saving the model.
- **prompt_encoder** ([PromptEncoder](/docs/peft/pr_3289/en/package_reference/p_tuning#peft.PromptEncoder)) -- The prompt encoder used for Peft if
using [PromptLearningConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PromptLearningConfig).
- **prompt_tokens** (`torch.Tensor`) -- The virtual prompt tokens used for Peft if
using [PromptLearningConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PromptLearningConfig).
- **transformer_backbone_name** (`str`) -- The name of the transformer
backbone in the base model if using [PromptLearningConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PromptLearningConfig).
- **word_embeddings** (`torch.nn.Embedding`) -- The word embeddings of the transformer backbone
in the base model if using [PromptLearningConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PromptLearningConfig).
- **adapter_name** (`str`) --
The name of the adapter to be added.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) --
The configuration of the adapter to be added.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device. Useful to speed up the process when loading saved
adapters. Don't use this option when creating a new PEFT adapter for training.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter
weights using float16 and bfloat16 to float32, as this is typically required for stable training, and
only affect select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the
corresponding layer.
Add an adapter to the model based on the passed configuration.
This adapter is not trained. To load a trained adapter, check out [PeftModel.load_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.load_adapter).
The name for the new adapter should be unique.
The new adapter is not automatically set as the active adapter. Use [PeftModel.set_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.set_adapter) to set the active
adapter.
Updates or create model card to include information about peft:
1. Adds `peft` library tag
2. Adds peft version
3. Adds base model info
4. Adds quantization information if it was used
- **adapter_name** (str) -- Name of the adapter to be deleted.
Deletes an existing adapter.
Context manager that disables the adapter module. Use this to run inference on the base model.
Example:
```py
>>> with model.disable_adapter():
... model(inputs)
```
Forward pass of the model.
- **model** (`torch.nn.Module`) --
The model to be adapted. For 🤗 Transformers models, the model should be initialized with the
[from_pretrained](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel.from_pretrained).
- **model_id** (`str` or `os.PathLike`) --
The name of the PEFT configuration to use. Can be either:
- A string, the `model id` of a PEFT configuration hosted inside a model repo on the Hugging Face
Hub.
- A path to a directory containing a PEFT configuration file saved using the `save_pretrained`
method (`./my_peft_config_directory/`).
- **adapter_name** (`str`, *optional*, defaults to `"default"`) --
The name of the adapter to be loaded. This is useful for loading multiple adapters.
- **is_trainable** (`bool`, *optional*, defaults to `False`) --
Whether the adapter should be trainable or not. If `False`, the adapter will be frozen and can only be
used for inference.
- **config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig), *optional*) --
The configuration object to use instead of an automatically loaded configuration. This configuration
object is mutually exclusive with `model_id` and `kwargs`. This is useful when configuration is already
loaded before calling `from_pretrained`.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter
weights using float16 and bfloat16 to float32, as this is typically required for stable training, and
only affect select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the
corresponding layer.
- **ephemeral_gpu_offload** (`bool`, *optional*) --
Whether to use ephemeral GPU offloading for partially loaded modules. Defaults to `False`. This is
useful when parts of the model and/or components (such as adapters) are kept in CPU memory until they
are needed. Rather than perform expensive operations on small data, the data is transferred to the GPU
on-demand, the operation(s) performed, and the results moved back to CPU memory. This brings a slight
momentary VRAM overhead but gives orders of magnitude speedup in certain cases.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device before loading the saved weights. Useful to speed up the
process.
- **torch_device** (`str`, *optional*, defaults to None) --
The device to load the adapter on. If `None`, the device will be inferred.
- **key_mapping** (dict, *optional*, defaults to None) --
Extra mapping of PEFT `state_dict` keys applied before loading the `state_dict`. When this mapping is
applied, the PEFT-specific `"base_model.model"` prefix is removed beforehand and the adapter name (e.g.
`"default"`) is not inserted yet. Only pass this argument if you know what you're doing.
- **kwargs** -- (`optional`):
Additional keyword arguments passed along to the specific PEFT configuration class.
Instantiate a PEFT model from a pretrained model and loaded PEFT weights.
Note that the passed `model` may be modified inplace.
Returns the base model.
- **model** ([~PeftModel](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel)) --
The model to get the adapter layer status from.list`peft.peft_model.TunerLayerStatus`A list of dataclasses, each containing the status of the corresponding adapter layer.
Get the status of each adapter layer in the model.
This method returns a list of `TunerLayerStatus` dataclass instances, each of which contains the following
attributes:
- `name` (`str`):
The name of the adapter layer, e.g. `model.encoder.block.0.layer.0.SelfAttention.q`.
- `module_type` (`str`):
The type of the adapter layer, e.g. `lora.Linear`.
- `enabled` (`bool`):
Whether the adapter layer is enabled.
- `active_adapters` (`list[str]`):
The names of the active adapters, if any, e.g. `["default"]`.
- `merged_adapters` (`list[str]`):
The names of the merged adapters, if any, e.g. `["default"]`.
- `available_adapters` (`list[str]`):
The names of the available adapters, e.g. `["default"]`.
- `quantization_backend` (`str` or `None`):
The name of the quantization backend, e.g. `"bnb 4bit"`, or `None` if not quantized.
- **model** ([~PeftModel](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel)) --
The model to get the adapter layer status from.`peft.peft_model.TunerModelStatus`A dataclass containing the status of the model.
Get the status of tuners of the model.
This method returns a `TunerModelStatus` dataclass instance, which contains the following attributes:
- `base_model_type` (`str`):
The type of the base model, e.g. `T5Model`.
- `adapter_model_type` (`str`):
The type of the adapter model, e.g. `LoraModel`.
- `peft_types` (`dict[str, str]`):
The mapping of adapter name to adapter type, e.g. `{"default": "LORA"}`.
- `trainable_params` (`int`):
The number of trainable parameters in the model.
- `total_params` (`int`):
The total number of parameters in the model.
- `num_adapter_layers` (`int`):
The number of adapter layers in the model.
- `enabled` (`bool`, `Literal["irregular"]`):
Whether all adapter layers are enabled. If some are enabled and some are not, this will be `"irregular"`.
This means that your model is in an inconsistent state and might not work as expected.
- `active_adapters` (`list[str]`, `Literal["irregular"]`):
The names of the active adapters. If the active adapters are not consistent across all layers, this will be
`"irregular"`, which means that your model is in an inconsistent state and might not work as expected.
- `merged_adapters` (`list[str]`, `Literal["irregular"]`):
The names of the merged adapters. If the merged adapters are not consistent across all layers, this will be
`"irregular"`, which means that your model is in an inconsistent state and might not work as expected.
- `available_adapters` (`list[str]`):
The names of the available adapters, e.g. `["default"]`.
- `quantization_backend` (`str`, `None`, `Literal["irregular"]`):
The name of the quantization backend, e.g. `"bnb 4bit"`, or `None` if not quantized. If the backend is not
consistent across all layers, this will be `"irregular"`.
Returns the number of trainable parameters and the number of all parameters in the model.
Returns the virtual prompts to use for Peft. Only applicable when using a prompt learning method.
Returns the prompt embedding to save when saving the model. Only applicable when using a prompt learning
method.
- **model_id** (`str` or `os.PathLike`) --
The name of the PEFT configuration to use. Can be either:
- A string, the `model id` of a PEFT configuration hosted inside a model repo on the Hugging Face
Hub.
- A path to a directory containing a PEFT configuration file saved using the `save_pretrained`
method (`./my_peft_config_directory/`).
- **adapter_name** (`str`) --
The name of the adapter to be added.
- **is_trainable** (`bool`, *optional*, defaults to `False`) --
Whether the adapter should be trainable or not. If `False`, the adapter will be frozen and can only be
used for inference.
- **torch_device** (`str`, *optional*, defaults to None) --
The device to load the adapter on. If `None`, the device will be inferred.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter
weights using float16 and bfloat16 to float32, as this is typically required for stable training, and
only affect select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the
corresponding layer.
- **ephemeral_gpu_offload** (`bool`, *optional*, defaults to `False`) --
Whether to use ephemeral GPU offloading for partially loaded modules. Defaults to `False`.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device before loading the saved weights. Useful to speed up the
process.
- **key_mapping** (dict, *optional*, defaults to None) --
Extra mapping of PEFT `state_dict` keys applied before loading the `state_dict`. When this mapping is
applied, the PEFT-specific `"base_model.model"` prefix is removed beforehand and the adapter name (e.g.
`"default"`) is not inserted yet. Only pass this argument if you know what you're doing.
- **kwargs** -- (`optional`):
Additional arguments to modify the way the adapter is loaded, e.g. the token for Hugging Face Hub.
Load a trained adapter into the model.
The name for the new adapter should be unique.
The new adapter is not automatically set as the active adapter. Use [PeftModel.set_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.set_adapter) to set the active
adapter.
Prepares the model for gradient checkpointing if necessary
Prints the number of trainable parameters in the model.
Note: print_trainable_parameters() uses get_nb_trainable_parameters() which is different from
num_parameters(only_trainable=True) from huggingface/transformers. get_nb_trainable_parameters() returns
(trainable parameters, all parameters) of the Peft Model which includes modified backbone transformer model.
For techniques like LoRA, the backbone transformer model is modified in place with LoRA modules. However, for
prompt tuning, the backbone transformer model is unmodified. num_parameters(only_trainable=True) returns number
of trainable parameters of the backbone transformer model which can be different.
- **save_directory** (`str`) --
Directory where the adapter model and configuration files will be saved (will be created if it does not
exist).
- **safe_serialization** (`bool`, *optional*) --
Whether to save the adapter files in safetensors format, defaults to `True`.
- **selected_adapters** (`List[str]`, *optional*) --
A list of adapters to be saved. If `None`, will default to all adapters.
- **save_embedding_layers** (`Union[bool, str]`, *optional*, defaults to `"auto"`) --
If `True`, save the embedding layers in addition to adapter weights. If `auto`, checks the common
embedding layers `peft.utils.other.EMBEDDING_LAYER_NAMES` in config's `target_modules` when available.
and automatically sets the boolean flag. This only works for 🤗 transformers models.
- **is_main_process** (`bool`, *optional*) --
Whether the process calling this is the main process or not. Will default to `True`. Will not save the
checkpoint if not on the main process, which is important for multi device setups (e.g. DDP).
- **path_initial_model_for_weight_conversion** (`str`, *optional*) --
The path to the initialized adapter, which is obtained after initializing the model with
PiSSA/CorDA/OLoRA and before performing any training. When `path_initial_model_for_weight_conversion`
is not None, the difference in adapter before and after fine-tuning is calculated. This difference can
be represented as the parameters of a standard LoRA adapter. In contrast to PiSSA and friends, using
this converted adapter does not require changes to the base model, thus conveniently allowing the use
of multiple PiSSA/CorDA/OLoRA adapters with LoRA adapters, and the activation or deactivation of any
adapters. Note that this conversion is not supported if `rslora` is used in combination with
`rank_pattern` or `alpha_pattern`. See [peft.tuners.lora.LoraModel.subtract_mutated_init()](/docs/peft/pr_3289/en/package_reference/lora#peft.LoraModel.subtract_mutated_init) for more
information.
- **kwargs** (additional keyword arguments, *optional*) --
Additional keyword arguments passed along to the `push_to_hub` method.
This function saves the adapter model and the adapter configuration files to a directory, so that it can be
reloaded using the [PeftModel.from_pretrained()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.from_pretrained) class method, and also used by the `PeftModel.push_to_hub()`
method.
- **adapter_name** (`str`) --
The name of the adapter to be set as active. The adapter must be loaded first.
- **inference_mode** (`bool`, optional) --
Whether the activated adapter should be frozen (i.e. `requires_grad=False`). Default is False.
Sets the active adapter.
Only one adapter can be active at a time.
Additionally, this function will set the specified adapter to trainable (i.e., requires_grad=True) unless
inference_mode is True.
- **adapter_names** (`str` or `Sequence[str]`) --
The name of the adapter(s) whose gradients should be enabled/disabled.
- **requires_grad** (`bool`, *optional*) --
Whether to enable (`True`, default) or disable (`False`).
Enable or disable gradients on the given adapter(s).
Note: Not supported for prompt learning methods like prompt tuning.
Whether it is possible for the adapter of this model to be converted to LoRA.
Normally, this works if the PEFT method is additive, i.e. W' = W_base + delta_weight.
## PeftModelForSequenceClassification[[peft.PeftModelForSequenceClassification]]
A `PeftModel` for sequence classification tasks.
- **model** ([PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel)) -- Base transformer model.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) -- Peft config.
- **adapter_name** (`str`, *optional*) -- The name of the adapter, defaults to `"default"`.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter weights
using float16 and bfloat16 to float32, as this is typically required for stable training, and only affect
select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the corresponding layer.
Peft model for sequence classification tasks.
**Attributes**:
- **config** (`PretrainedConfig`) -- The configuration object of the base model.
- **cls_layer_name** (`str`) -- The name of the classification layer.
Example:
```py
>>> from transformers import AutoModelForSequenceClassification
>>> from peft import PeftModelForSequenceClassification, get_peft_config
>>> config = {
... "peft_type": "PREFIX_TUNING",
... "task_type": "SEQ_CLS",
... "inference_mode": False,
... "num_virtual_tokens": 20,
... "token_dim": 768,
... "num_transformer_submodules": 1,
... "num_attention_heads": 12,
... "num_layers": 12,
... "encoder_hidden_size": 768,
... "prefix_projection": False,
... "postprocess_past_key_value_function": None,
... }
>>> peft_config = get_peft_config(config)
>>> model = AutoModelForSequenceClassification.from_pretrained("bert-base-cased")
>>> peft_model = PeftModelForSequenceClassification(model, peft_config)
>>> peft_model.print_trainable_parameters()
trainable params: 370178 || all params: 108680450 || trainable%: 0.3406113979101117
```
## PeftModelForTokenClassification[[peft.PeftModelForTokenClassification]]
A `PeftModel` for token classification tasks.
- **model** ([PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel)) -- Base transformer model.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) -- Peft config.
- **adapter_name** (`str`, *optional*) -- The name of the adapter, defaults to `"default"`.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter weights
using float16 and bfloat16 to float32, as this is typically required for stable training, and only affect
select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the corresponding layer.
Peft model for token classification tasks.
**Attributes**:
- **config** (`PretrainedConfig`) -- The configuration object of the base model.
- **cls_layer_name** (`str`) -- The name of the classification layer.
Example:
```py
>>> from transformers import AutoModelForSequenceClassification
>>> from peft import PeftModelForTokenClassification, get_peft_config
>>> config = {
... "peft_type": "PREFIX_TUNING",
... "task_type": "TOKEN_CLS",
... "inference_mode": False,
... "num_virtual_tokens": 20,
... "token_dim": 768,
... "num_transformer_submodules": 1,
... "num_attention_heads": 12,
... "num_layers": 12,
... "encoder_hidden_size": 768,
... "prefix_projection": False,
... "postprocess_past_key_value_function": None,
... }
>>> peft_config = get_peft_config(config)
>>> model = AutoModelForTokenClassification.from_pretrained("bert-base-cased")
>>> peft_model = PeftModelForTokenClassification(model, peft_config)
>>> peft_model.print_trainable_parameters()
trainable params: 370178 || all params: 108680450 || trainable%: 0.3406113979101117
```
- **adapter_name** (`str`) --
The name of the adapter to be added.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) --
The configuration of the adapter to be added.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device. Useful to speed up the process when loading saved
adapters. Don't use this option when creating a new PEFT adapter for training.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter
weights using float16 and bfloat16 to float32, as this is typically required for stable training, and
only affect select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the
corresponding layer.
Add an adapter to the model based on the passed configuration.
This adapter is not trained. To load a trained adapter, check out [PeftModel.load_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.load_adapter).
The name for the new adapter should be unique.
The new adapter is not automatically set as the active adapter. Use [PeftModel.set_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.set_adapter) to set the active
adapter.
## PeftModelForCausalLM[[peft.PeftModelForCausalLM]]
A `PeftModel` for causal language modeling.
- **model** ([PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel)) -- Base transformer model.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) -- Peft config.
- **adapter_name** (`str`, *optional*) -- The name of the adapter, defaults to `"default"`.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter weights
using float16 and bfloat16 to float32, as this is typically required for stable training, and only affect
select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the corresponding layer.
Peft model for causal language modeling.
Example:
```py
>>> from transformers import AutoModelForCausalLM
>>> from peft import PeftModelForCausalLM, get_peft_config
>>> config = {
... "peft_type": "PREFIX_TUNING",
... "task_type": "CAUSAL_LM",
... "inference_mode": False,
... "num_virtual_tokens": 20,
... "token_dim": 1280,
... "num_transformer_submodules": 1,
... "num_attention_heads": 20,
... "num_layers": 36,
... "encoder_hidden_size": 1280,
... "prefix_projection": False,
... "postprocess_past_key_value_function": None,
... }
>>> peft_config = get_peft_config(config)
>>> model = AutoModelForCausalLM.from_pretrained("gpt2-large")
>>> peft_model = PeftModelForCausalLM(model, peft_config)
>>> peft_model.print_trainable_parameters()
trainable params: 1843200 || all params: 775873280 || trainable%: 0.23756456724479544
```
## PeftModelForSeq2SeqLM[[peft.PeftModelForSeq2SeqLM]]
A `PeftModel` for sequence-to-sequence language modeling.
- **model** ([PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel)) -- Base transformer model.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) -- Peft config.
- **adapter_name** (`str`, *optional*) -- The name of the adapter, defaults to `"default"`.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter weights
using float16 and bfloat16 to float32, as this is typically required for stable training, and only affect
select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the corresponding layer.
Peft model for sequence-to-sequence language modeling.
Example:
```py
>>> from transformers import AutoModelForSeq2SeqLM
>>> from peft import PeftModelForSeq2SeqLM, get_peft_config
>>> config = {
... "peft_type": "LORA",
... "task_type": "SEQ_2_SEQ_LM",
... "inference_mode": False,
... "r": 8,
... "target_modules": ["q", "v"],
... "lora_alpha": 32,
... "lora_dropout": 0.1,
... "fan_in_fan_out": False,
... "enable_lora": None,
... "bias": "none",
... }
>>> peft_config = get_peft_config(config)
>>> model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
>>> peft_model = PeftModelForSeq2SeqLM(model, peft_config)
>>> peft_model.print_trainable_parameters()
trainable params: 884736 || all params: 223843584 || trainable%: 0.3952474242013566
```
## PeftModelForQuestionAnswering[[peft.PeftModelForQuestionAnswering]]
A `PeftModel` for question answering.
- **model** ([PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel)) -- Base transformer model.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) -- Peft config.
- **adapter_name** (`str`, *optional*) -- The name of the adapter, defaults to `"default"`.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter weights
using float16 and bfloat16 to float32, as this is typically required for stable training, and only affect
select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the corresponding layer.
Peft model for extractive question answering.
**Attributes**:
- **config** (`PretrainedConfig`) -- The configuration object of the base model.
- **cls_layer_name** (`str`) -- The name of the classification layer.
Example:
```py
>>> from transformers import AutoModelForQuestionAnswering
>>> from peft import PeftModelForQuestionAnswering, get_peft_config
>>> config = {
... "peft_type": "LORA",
... "task_type": "QUESTION_ANS",
... "inference_mode": False,
... "r": 16,
... "target_modules": ["query", "value"],
... "lora_alpha": 32,
... "lora_dropout": 0.05,
... "fan_in_fan_out": False,
... "bias": "none",
... }
>>> peft_config = get_peft_config(config)
>>> model = AutoModelForQuestionAnswering.from_pretrained("bert-base-cased")
>>> peft_model = PeftModelForQuestionAnswering(model, peft_config)
>>> peft_model.print_trainable_parameters()
trainable params: 592900 || all params: 108312580 || trainable%: 0.5473971721475013
```
- **adapter_name** (`str`) --
The name of the adapter to be added.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) --
The configuration of the adapter to be added.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device. Useful to speed up the process when loading saved
adapters. Don't use this option when creating a new PEFT adapter for training.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter
weights using float16 and bfloat16 to float32, as this is typically required for stable training, and
only affect select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the
corresponding layer.
Add an adapter to the model based on the passed configuration.
This adapter is not trained. To load a trained adapter, check out [PeftModel.load_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.load_adapter).
The name for the new adapter should be unique.
The new adapter is not automatically set as the active adapter. Use [PeftModel.set_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.set_adapter) to set the active
adapter.
## PeftModelForFeatureExtraction[[peft.PeftModelForFeatureExtraction]]
A `PeftModel` for getting extracting features/embeddings from transformer models.
- **model** ([PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel)) -- Base transformer model.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) -- Peft config.
- **adapter_name** (`str`, *optional*) -- The name of the adapter, defaults to `"default"`.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter weights
using float16 and bfloat16 to float32, as this is typically required for stable training, and only affect
select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the corresponding layer.
Peft model for extracting features/embeddings from transformer models
**Attributes**:
- **config** (`PretrainedConfig`) -- The configuration object of the base model.
Example:
```py
>>> from transformers import AutoModel
>>> from peft import PeftModelForFeatureExtraction, get_peft_config
>>> config = {
... "peft_type": "LORA",
... "task_type": "FEATURE_EXTRACTION",
... "inference_mode": False,
... "r": 16,
... "target_modules": ["query", "value"],
... "lora_alpha": 32,
... "lora_dropout": 0.05,
... "fan_in_fan_out": False,
... "bias": "none",
... }
>>> peft_config = get_peft_config(config)
>>> model = AutoModel.from_pretrained("bert-base-cased")
>>> peft_model = PeftModelForFeatureExtraction(model, peft_config)
>>> peft_model.print_trainable_parameters()
```
## PeftMixedModel[[peft.PeftMixedModel]]
A `PeftModel` for mixing different adapter types (e.g. LoRA and LoHa).
- **model** (`torch.nn.Module`) --
The model to be tuned.
- **config** (`PeftConfig`) --
The config of the model to be tuned. The adapter type must be compatible.
- **adapter_name** (`str`, `optional`, defaults to `"default"`) --
The name of the first adapter.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device. Useful to speed up the loading process.
PeftMixedModel for loading mixing different types of adapters for inference.
This class does not support loading/saving, and it shouldn't usually be initialized directly. Instead, use
`get_peft_model` with the argument `mixed=True`.
> [!TIP] > Read the [Mixed adapter types](https://huggingface.co/docs/peft/en/developer_guides/mixed_models) guide
to learn > more about using different adapter types.
Example:
```py
>>> base_model = ... # load the base model, e.g. from transformers
>>> peft_model = PeftMixedModel.from_pretrained(base_model, path_to_adapter1, "adapter1").eval()
>>> peft_model.load_adapter(path_to_adapter2, "adapter2")
>>> peft_model.set_adapter(["adapter1", "adapter2"]) # activate both adapters
>>> peft_model(data) # forward pass using both adapters
```
- **adapter_name** (`str`) --
The name of the adapter to be added.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) --
The configuration of the adapter to be added.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device. Useful to speed up the process when loading saved
adapters.
> [!TIP] > Don't use `low_cpu_mem_usage=True` when creating a new PEFT adapter for training (training
is untested > and discouraged for PeftMixedModel in general).
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter
weights using float16 and bfloat16 to float32, as this is typically required for stable training, and
only affect select PEFT tuners. If set to `False`, the dtypes will stay the same as those of the
corresponding layer.
Add an adapter to the model based on the passed configuration.
This adapter is not trained. To load a trained adapter, check out [PeftModel.load_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.load_adapter).
The name for the new adapter should be unique.
The new adapter is not automatically set as the active adapter. Use [PeftModel.set_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.set_adapter) to set the active
adapter.
Disables the adapter module.
Forward pass of the model.
- **model** (`nn.Module`) --
The model to be adapted.
- **model_id** (`str` or `os.PathLike`) --
The name of the PEFT configuration to use. Can be either:
- A string, the `model id` of a PEFT configuration hosted inside a model repo on the Hugging Face
Hub.
- A path to a directory containing a PEFT configuration file saved using the `save_pretrained`
method (`./my_peft_config_directory/`).
- **adapter_name** (`str`, *optional*, defaults to `"default"`) --
The name of the adapter to be loaded. This is useful for loading multiple adapters.
- **is_trainable** (`bool`, *optional*, defaults to `False`) --
Whether the adapter should be trainable or not. If `False`, the adapter will be frozen and use for
inference
- **config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig), *optional*) --
The configuration object to use instead of an automatically loaded configuration. This configuration
object is mutually exclusive with `model_id` and `kwargs`. This is useful when configuration is already
loaded before calling `from_pretrained`.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device before loading the saved weights. Useful to speed up the
process.
- **kwargs** -- (`optional`):
Additional keyword arguments passed along to the specific PEFT configuration class.
Instantiate a PEFT mixed model from a pretrained model and loaded PEFT weights.
Note that the passed `model` may be modified inplace.
Generate output.
Returns the number of trainable parameters and number of all parameters in the model.
- **adapter_name** (`str`) --
The name of the adapter to be added.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) --
The configuration of the adapter to be added.
- **is_trainable** (`bool`, *optional*, defaults to `False`) --
Whether the adapter should be trainable or not. If `False`, the adapter will be frozen and can only be
used for inference.
- **torch_device** (`str`, *optional*, defaults to None) --
The device to load the adapter on. If `None`, the device will be inferred.
- **autocast_adapter_dtype** (`bool`, *optional*, defaults to `True`) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter
weights using float16 and bfloat16 to float32, as this is typically required for stable training, and
only affect select PEFT tuners.
- **ephemeral_gpu_offload** (`bool`, *optional*, defaults to `False`) --
Whether to use ephemeral GPU offloading for partially loaded modules. Defaults to `False`.
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device before loading the saved weights. Useful to speed up the
process.
- **kwargs** -- (`optional`):
Additional arguments to modify the way the adapter is loaded, e.g. the token for Hugging Face Hub.
Load a trained adapter into the model.
The name for the new adapter should be unique.
The new adapter is not automatically set as the active adapter. Use [PeftModel.set_adapter()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel.set_adapter) to set the active
adapter.
- **progressbar** (`bool`) --
whether to show a progressbar indicating the unload and merge process
- **safe_merge** (`bool`) --
whether to activate the safe merging check to check if there is any potential Nan in the adapter
weights
- **adapter_names** (`List[str]`, *optional*) --
The list of adapter names that should be merged. If None, all active adapters will be merged. Defaults
to `None`.
This method merges the adapter layers into the base model. This is needed if someone wants to use the base
model as a standalone model.
Prints the number of trainable parameters in the model.
Note: print_trainable_parameters() uses get_nb_trainable_parameters() which is different from
num_parameters(only_trainable=True) from huggingface/transformers. get_nb_trainable_parameters() returns
(trainable parameters, all parameters) of the Peft Model which includes modified backbone transformer model.
For techniques like LoRA, the backbone transformer model is modified in place with LoRA modules. However, for
prompt tuning, the backbone transformer model is unmodified. num_parameters(only_trainable=True) returns number
of trainable parameters of the backbone transformer model which can be different.
- **adapter_name** (str, list[str]) --
The name(s) of the adapter(s) to set as active
- **inference_mode** (bool, optional) --
Whether the activated adapter should be frozen (i.e. `requires_grad=False`). Default is False.
Sets the active adapter(s) for the model.
Note that the order in which the adapters are applied during the forward pass may not be the same as the order
in which they are passed to this function. Instead, the order during the forward pass is determined by the
order in which the adapters were loaded into the model. The active adapters only determine which adapters are
active during the forward pass, but not the order in which they are applied.
Additionally, this function will set the specified adapter to trainable (i.e., requires_grad=True) unless
inference_mode is True.
Gets back the base model by removing all the adapter modules without merging. This gives back the original base
model.
## Utilities[[peft.cast_mixed_precision_params]]
- **model** (`torch.nn.Module`) --
The model to cast the non-trainable parameters of.
- **dtype** (`torch.dtype`) --
The dtype to cast the non-trainable parameters to. The `dtype` can be `torch.float16` or
Cast all non-trainable parameters of the model to the given `dtype`. The `dtype` can be `torch.float16` or
`torch.bfloat16` as per the mixed-precision training you are performing. The trainable parameters are cast to full
precision. This is meant to reduce the GPU memory usage when using PEFT methods by using half-precision dtype for
non-trainable parameters. Having the trainable parameters in full-precision preserves training stability when using
automatic mixed-precision training.
`torch.bfloat16` as per the mixed-precision training you are performing.
- **model** ([transformers.PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel)) --
Model to be wrapped.
- **peft_config** ([PeftConfig](/docs/peft/pr_3289/en/package_reference/config#peft.PeftConfig)) --
Configuration object containing the parameters of the Peft model.
- **adapter_name** (`str`, `optional`, defaults to `"default"`) --
The name of the adapter to be injected, if not provided, the default adapter name is used ("default").
- **mixed** (`bool`, `optional`, defaults to `False`) --
Whether to allow mixing different (compatible) adapter types.
- **autocast_adapter_dtype** (`bool`, *optional*) --
Whether to autocast the adapter dtype. Defaults to `True`. Right now, this will only cast adapter weights
using float16 or bfloat16 to float32, as this is typically required for stable training, and only affect
select PEFT tuners.
- **revision** (`str`, `optional`, defaults to `main`) --
The revision of the base model. If this isn't set, the saved peft model will load the `main` revision for
the base model
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device. Useful to speed up the loading process. Leave this setting as
False if you intend on training the model, unless the adapter weights will be replaced by different weights
before training starts.
Returns a Peft model object from a model and a config, where the model will be modified in-place.
- **peft_config** (`PeftConfig`) --
Configuration object containing the parameters of the PEFT model.
- **model** (`torch.nn.Module`) --
The input model where the adapter will be injected.
- **adapter_name** (`str`, `optional`, defaults to `"default"`) --
The name of the adapter to be injected, if not provided, the default adapter name is used ("default").
- **low_cpu_mem_usage** (`bool`, `optional`, defaults to `False`) --
Create empty adapter weights on meta device. Useful to speed up the loading process.
- **state_dict** (`dict`, *optional*, defaults to `None`) --
If a `state_dict` is passed here, the adapters will be injected based on the entries of the state_dict.
This can be useful when the exact `target_modules` of the PEFT method is unknown, for instance because the
checkpoint was created without meta data. Note that the values from the `state_dict` are not used, only the
keys are used to determine the correct layers that should be adapted.
Create PEFT layers and inject them into the model in-place.
Currently the API does not support prompt learning methods and adaption prompt.
This function is similar to [get_peft_model()](/docs/peft/pr_3289/en/package_reference/peft_model#peft.get_peft_model) but it does not return a [PeftModel](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel) instance. Instead, it returns
the original, mutated instance of the passed model.
- **model** ([PeftModel](/docs/peft/pr_3289/en/package_reference/peft_model#peft.PeftModel)) -- The Peft model. When using torch.nn.DistributedDataParallel, DeepSpeed or FSDP,
the model should be the underlying model/unwrapped model (i.e. model.module).
- **state_dict** (`dict`, *optional*, defaults to `None`) --
The state dict of the model. If not provided, the state dict of the passed model will be used.
- **adapter_name** (`str`, *optional*, defaults to `"default"`) --
The name of the adapter whose state dict should be returned.
- **unwrap_compiled** (`bool`, *optional*, defaults to `False`) --
Whether to unwrap the model if torch.compile was used.
- **save_embedding_layers** (`Union[bool, str]`, , *optional*, defaults to `auto`) --
If `True`, save the embedding layers in addition to adapter weights. If `auto`, checks the common embedding
layers `peft.utils.other.EMBEDDING_LAYER_NAMES` in config's `target_modules` when available. Based on it
sets the boolean flag. This only works for 🤗 transformers models.
Get the state dict of the given adapter of the PEFT model.
This only includes the PEFT parameters, not the parameters of the base model. Thus the returned `state_dict` is
generally small compared to the full model size. To retrieve the full `state_dict`, just call `model.state_dict()`.
Note that the adapter name is removed from the `state_dict`, as this is just an arbitrary name that can be changed
when loading the adapter. So e.g. if the adapter name is `'default'` and the original key is
`'model.q_proj.lora_A.default.weight'`, the returned key will be `'model.q_proj.lora_A.weight'`. Use this function
in conjunction with [set_peft_model_state_dict()](/docs/peft/pr_3289/en/package_reference/functional#peft.set_peft_model_state_dict) to take care of the adapter name when loading weights.
- **model** (`transformers.PreTrainedModel`) --
The loaded model from `transformers`
- **use_gradient_checkpointing** (`bool`, *optional*, defaults to `True`) --
If True, use gradient checkpointing to save memory at the expense of slower backward pass.
- **gradient_checkpointing_kwargs** (`dict`, *optional*, defaults to `None`) --
Keyword arguments to pass to the gradient checkpointing function, please refer to the documentation of
`torch.utils.checkpoint.checkpoint` for more details about the arguments that you can pass to that method.
Note this is only available in the latest transformers versions (> 4.34.1).
Note this method only works for `transformers` models.
This method wraps the entire protocol for preparing a model before running a training. This includes:
1- Cast the layernorm in fp32 2- making output embedding layer require grads 3- Add the upcasting of the lm
head to fp32 4- Freezing the base model layers to ensure they are not updated during training
- **model** ([Union[`~PeftModel`, `~transformers.PreTrainedModel`, `nn.Module`]]) --
The model to get the adapter layer status from.list`peft.peft_model.TunerLayerStatus`A list of dataclasses, each containing the status of the corresponding adapter layer.
Get the status of each adapter layer in the model.
This function returns a list of `TunerLayerStatus` dataclass instances, each of which contains the following
attributes:
- `name` (`str`):
The name of the adapter layer, e.g. `model.encoder.block.0.layer.0.SelfAttention.q`.
- `module_type` (`str`):
The type of the adapter layer, e.g. `lora.Linear`.
- `enabled` (`bool`):
Whether the adapter layer is enabled.
- `active_adapters` (`list[str]`):
The names of the active adapters, if any, e.g. `["default"]`.
- `merged_adapters` (`list[str]`):
The names of the merged adapters, if any, e.g. `["default"]`.
- requires_grad : dict[str, bool | Literal["irregular"]]
The requires_grad status of the parameters for each adapter module. Ideally, it should be either `True` or
`False`. If the requires_grad status is not consistent across all parameters, the value will be set to
`"irregular"`.
- `available_adapters` (`list[str]`):
The names of the available adapters, e.g. `["default"]`.
- `devices` (`dict[str, list[str]]`):
The devices where the parameters of the given adapter are stored, e.g. `["cuda"]`.
- `quantization_backend` (`str` or `None`):
The name of the quantization backend, e.g. `"bnb 4bit"`, or `None` if not quantized.
- **model** ([Union[`~PeftModel`, `~transformers.PreTrainedModel`, `nn.Module`]]) --
The model to get the adapter layer status from.`peft.peft_model.TunerModelStatus`A dataclass containing the status of the model.
Get the status of tuners of the model.
This function returns a `TunerModelStatus` dataclass instance, which contains the following attributes:
- `base_model_type` (`str`):
The type of the base model, e.g. `T5Model`.
- `adapter_model_type` (`str`):
The type of the adapter model, e.g. `LoraModel`.
- `peft_types` (`dict[str, str]`):
The mapping of adapter name to adapter type, e.g. `{"default": "LORA"}`.
- `trainable_params` (`int`):
The number of trainable parameters in the model.
- `total_params` (`int`):
The total number of parameters in the model.
- `num_adapter_layers` (`int`):
The number of adapter layers in the model.
- `enabled` (`bool`, `Literal["irregular"]`):
Whether all adapter layers are enabled. If some are enabled and some are not, this will be `"irregular"`. This
means that your model is in an inconsistent state and might not work as expected.
- `active_adapters` (`list[str]`, `Literal["irregular"]`):
The names of the active adapters. If the active adapters are not consistent across all layers, this will be
`"irregular"`, which means that your model is in an inconsistent state and might not work as expected.
- `merged_adapters` (`list[str]`, `Literal["irregular"]`):
The names of the merged adapters. If the merged adapters are not consistent across all layers, this will be
`"irregular"`, which means that your model is in an inconsistent state and might not work as expected.
- `requires_grad` (`dict[str, bool | Literal["irregular"]]`):
Whether for the given adapter, all adapter layers have `requires_grad` set to `True` or `False`. If there is a
mix, this will be set to `"irregular"`, which means that your model is in an inconsistent state and might not
work as expected.
- `available_adapters` (`list[str]`):
The names of the available adapters, e.g. `["default"]`.
- `devices` (`dict[str, list[str]]`):
The devices where the parameters of the given adapter are stored, e.g. `["cuda"]`.
- `quantization_backend` (`str`, `None`, `Literal["irregular"]`):
The name of the quantization backend, e.g. `"bnb 4bit"`, or `None` if not quantized. If the backend is not
consistent across all layers, this will be `"irregular"`.

Xet Storage Details

Size:
50 kB
·
Xet hash:
ce8d253cc94e7a1bbec04805a3328f5478cfbe2a04e8217e1e5ab1a4c94d85db

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.