Skip to content

Commit

Permalink
style
Browse files Browse the repository at this point in the history
  • Loading branch information
calpt committed Jan 18, 2025
1 parent e964973 commit 23c6979
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/adapters/methods/reft.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ def __init__(
if dtype in [torch.float16, torch.bfloat16]:
warnings.warn(
"Orthogonal parametrization is not supported for half precision dtypes. Converting REFT projection layer to float32.",
UserWarning
UserWarning,
)
projection = projection.to(dtype=torch.float32)
self.projection = nn.utils.parametrizations.orthogonal(projection)
Expand Down

0 comments on commit 23c6979

Please sign in to comment.