Skip to content

Commit

Permalink
DOC Describe torch_device in from_pretrained docs (#1843)
Browse files Browse the repository at this point in the history
  • Loading branch information
BenjaminBossan committed Jun 10, 2024
1 parent 683db0f commit a8286a7
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/peft/peft_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -379,6 +379,8 @@ def from_pretrained(
loaded before calling `from_pretrained`.
autocast_adapter_dtype (`bool`, *optional*):
Whether to autocast the adapter dtype. Defaults to `True`. Only relevant for specific adapter types.
torch_device (`str`, *optional*, defaults to None):
The device to load the adapter on. If `None`, the device will be inferred.
kwargs: (`optional`):
Additional keyword arguments passed along to the specific PEFT configuration class.
"""
Expand Down

0 comments on commit a8286a7

Please sign in to comment.