add liger kernels

turbo_alignment/cherry_picks/__init__.py

@@ -1,5 +1,5 @@
 from .chat import ChatCherryPickCallback
 from .classification import ClassificationCherryPickCallback
-from .multimodal import MultimodalCherryPickCallback
+# from .multimodal import MultimodalCherryPickCallback
 from .rag import RagCherryPickCallback
 from .rm import RmCherryPickCallback

turbo_alignment/common/data/multimodal/audio/imagebind.py

@@ -1,10 +1,16 @@
 from pathlib import Path
 
 import torch
-import torchaudio
+try:
+    import torchaudio
+except:
+    ...
 from loguru import logger
-from pytorchvideo.data.clip_sampling import ConstantClipsPerVideoSampler
-from torchvision import transforms
+try:
+    from pytorchvideo.data.clip_sampling import ConstantClipsPerVideoSampler
+    from torchvision import transforms
+except:
+    ...
 
 from turbo_alignment.common.data.multimodal.image.base import BaseImageReader
 from turbo_alignment.common.data.multimodal.registry import AudioModalityReaderRegistry

turbo_alignment/common/data/multimodal/audio/whisper.py

@@ -1,12 +1,14 @@
 from pathlib import Path
+try:
+    import soundfile as sf
+    import torch
+    from transformers import AutoFeatureExtractor
 
-import soundfile as sf
-import torch
-from transformers import AutoFeatureExtractor
-
-from turbo_alignment.common.data.multimodal.image.base import BaseImageReader
-from turbo_alignment.common.data.multimodal.registry import AudioModalityReaderRegistry
-from turbo_alignment.settings.modality import ModalityReader
+    from turbo_alignment.common.data.multimodal.image.base import BaseImageReader
+    from turbo_alignment.common.data.multimodal.registry import AudioModalityReaderRegistry
+    from turbo_alignment.settings.modality import ModalityReader
+except:
+    ...
 
 
 @AudioModalityReaderRegistry.register(ModalityReader.WHISPER)

turbo_alignment/common/data/multimodal/image/imagebind.py

@@ -1,11 +1,14 @@
 from pathlib import Path
 
-import cv2
-import numpy as np
-import torch
-import torchvision.transforms as transforms
-import torchvision.transforms.functional as F
-
+try:
+    import cv2
+    import numpy as np
+    import torch
+    import torchvision.transforms as transforms
+    import torchvision.transforms.functional as F
+except:
+    ...
+
 from turbo_alignment.common.data.multimodal.image.base import BaseImageReader
 from turbo_alignment.common.data.multimodal.registry import ImageModalityReaderRegistry
 from turbo_alignment.settings.modality import ModalityReader