Fix path

prismer/dataset/caption_dataset.py

@@ -7,7 +7,7 @@
 import glob
 
 from torch.utils.data import Dataset
-from prismer.dataset.utils import *
+from dataset.utils import *
 from PIL import ImageFile
 ImageFile.LOAD_TRUNCATED_IMAGES = True
 

prismer/dataset/utils.py

@@ -13,7 +13,7 @@ import numpy as np
 import torchvision.transforms as transforms
 import torchvision.transforms.functional as transforms_f
 import pathlib
-from prismer.dataset.randaugment import RandAugment
+from dataset.randaugment import RandAugment
 
 
 cur_dir = pathlib.Path(__file__).parent

prismer/dataset/vqa_dataset.py

@@ -6,6 +6,8 @@
 
 from torch.utils.data import Dataset
 from dataset.utils import *
+from PIL import ImageFile
+ImageFile.LOAD_TRUNCATED_IMAGES = True
 
 
 class VQA(Dataset):

prismer_model.py

@@ -33,9 +33,7 @@ def download_models() -> None:
 
 
 def build_deformable_conv() -> None:
-    subprocess.run(
-        shlex.split('sh make.sh'),
-        cwd='prismer/experts/segmentation/mask2former/modeling/pixel_decoder/ops')
+    subprocess.run( shlex.split('sh make.sh'), cwd='prismer/experts/segmentation/mask2former/modeling/pixel_decoder/ops')
 
 
 def run_experts(image_path: str) -> tuple[str | None, ...]: