-
Notifications
You must be signed in to change notification settings - Fork 2.1k
Open
Labels
Description
In transformers, the torch_dtype
argument is deprecated in favor of dtype
(as in: AutoModel.from_pretrained(..., torch_dtype=torch.float16)
). Although it won't be removed anytime soon, it's still annoying to see the deprecation message. Therefore, PEFT should be updated to use dtype
instead of torch_dtype
.
After a quick search, the PEFT code proper is not affected, but there are instances in docs, examples, and tests. In general, a simple search and replace should do the job of updating the argument name.