test: sam

app.py

@@ -9,6 +9,13 @@ from controlnet_union import ControlNetModel_Union
 from pipeline_fill_sd_xl import StableDiffusionXLFillPipeline
 from PIL import Image, ImageDraw
 import numpy as np
+from sam2.sam2_image_predictor import SAM2ImagePredictor
+
+predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2.1-hiera-large")
+print(predictor)
+# with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
+#     predictor.set_image(<your_image>)
+#     masks, _, _ = predictor.predict(<input_prompts>)
 
 MODELS = {
     "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",