Fix issue with import

osrt/encoder.py

@@ -16,12 +16,12 @@ from osrt.layers import RayEncoder, Transformer, SlotAttention
 from osrt.utils.common import batch_iterator, MaskData, calculate_stability_score, get_indices_sorted_pos, get_positional_embedding, create_points_grid, rotation_matrix_to_euler_angles
 from osrt.utils.nerf import get_extrinsic
 
-from sam import sam_model_registry
-from sam.image_encoder import ImageEncoderViT
-from sam.mask_decoder import MaskDecoder
-from sam.prompt_encoder import PromptEncoder
-from sam.utils.transforms import ResizeLongestSide
-from sam.utils.amg import batched_mask_to_box, remove_small_regions, mask_to_rle_pytorch, area_from_rle
+from osrt.sam import sam_model_registry
+from osrt.sam.image_encoder import ImageEncoderViT
+from osrt.sam.mask_decoder import MaskDecoder
+from osrt.sam.prompt_encoder import PromptEncoder
+from osrt.sam.utils.transforms import ResizeLongestSide
+from osrt.sam.utils.amg import batched_mask_to_box, remove_small_regions, mask_to_rle_pytorch, area_from_rle
 
 from torch.nn.utils.rnn import pad_sequence