app

app.py

@@ -20,7 +20,7 @@ print("path", os.getcwd())
 model_type = "vit_h"
 predictor = SamPredictor(sam_model_registry[model_type](checkpoint="./GroundingDINO/weights/sam_vit_h_4b8939.pth").to(device))
 # ------Stable Diffusion
-pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting", torch_dtype=torch.float16,).to(device)
+pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting", torch_dtype=torch.float32).to(device)
 # ----Grounding DINO
 groundingdino_model = load_model("./GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py", "./GroundingDINO/weights/groundingdino_swint_ogc.pth")
 

requirements.txt

@@ -1,4 +1,5 @@
 segment-anything
+--extra-index-url https://download.pytorch.org/whl/cu113
 torch
 numpy
 opencv-python
@@ -6,4 +7,4 @@ torchvision
 diffusers
 supervision
 transformers
-addict
+addict

setup.sh

@@ -2,8 +2,7 @@
 
 # Clone the GroundingDINO repository
 # git clone https://github.com/IDEA-Research/GroundingDINO.git
-
-# Change directory to GroundingDINO
+--extra-index-url https://download.pytorch.org/whl/cu113
 cd GroundingDINO
 
 # Install the Python package