Merge branch 'master' into huggingface

modules/constants.py

@@ -7,3 +7,4 @@ IMAGE_FILE_EXT = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp']
 VIDEO_FILE_EXT = ['.mp4', '.avi', '.mov', '.wmv', '.flv', '.webm', '.mkv', '.mpeg', '.mpg', '.m4v', '.3gp', '.ts', '.vob']
 DEFAULT_COLOR = "#00FF00"
 DEFAULT_PIXEL_SIZE = 8
+

modules/sam_inference.py

@@ -41,7 +41,7 @@ class SamInference:
         self.model_dir = model_dir
         self.output_dir = output_dir
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+        self.dtype = torch.float16 if torch.cuda.is_available() else torch.bfloat16
         self.mask_generator = None
         self.image_predictor = None
         self.video_predictor = None