Skip to content

Commit a2e5c57

Browse files
committed
Fix type hint for peft_config parameter in TimmViT class
1 parent 9734e2d commit a2e5c57

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmlearn/modules/encoders/vision.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def __init__(
6363
pretrained: bool = True,
6464
freeze_layers: Union[int, float, list[int], bool] = False,
6565
freeze_layer_norm: bool = True,
66-
peft_config: Optional[PeftConfig] = None,
66+
peft_config: Optional["PeftConfig"] = None,
6767
model_kwargs: Optional[dict[str, Any]] = None,
6868
) -> None:
6969
super().__init__()

0 commit comments

Comments
 (0)