Refactor type imports for conditional use of peft package

Author: fcogidi

mmlearn/modules/encoders/vision.py

@@ -2,12 +2,11 @@
 
 import math
 from functools import partial
-from typing import Any, Callable, Literal, Optional, Union, cast
+from typing import TYPE_CHECKING, Any, Callable, Literal, Optional, Union, cast
 
 import timm
 import torch
 from hydra_zen import store
-from peft import PeftConfig
 from timm.models.vision_transformer import VisionTransformer as TimmVisionTransformer
 from timm.models.vision_transformer import global_pool_nlc
 from torch import nn
@@ -21,6 +20,10 @@
 from mmlearn.modules.layers.transformer_block import Block
 
 
+if TYPE_CHECKING:
+    from peft import PeftConfig
+
+
 @store(
     group="modules/encoders",
     provider="mmlearn",