update

DPT/requirements.txt

@@ -1,4 +1,4 @@
-torch==1.8.1
-torchvision==0.9.1
+#torch==1.8.1
+#torchvision==0.9.1
 opencv-python==4.5.2.54
 timm==0.4.5

demo_gradio.py

@@ -42,8 +42,10 @@ base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
 image_encoder_path = "models/image_encoder"
 ip_ckpt = "sdxl_models/ip-adapter_sdxl_vit-h.bin"
 controlnet_path = "diffusers/controlnet-depth-sdxl-1.0"
-device = "cuda"
-torch.cuda.empty_cache()
+#device = "cuda"
+#torch.cuda.empty_cache()
+device = devicetorch.device(torch)
+devicetorch.empty_cache(torch)
 
 # load SDXL pipeline
 controlnet = ControlNetModel.from_pretrained(controlnet_path, variant="fp16", use_safetensors=True, torch_dtype=torch.float16).to(device)

requirements.txt

@@ -7,5 +7,5 @@ opencv-python==4.7.0.68
 gradio==4.15.0
 accelerate==0.26.1
 git+https://github.com/fffiloni/pytorch-image-models.git
-torch==2.0.1
-torchvision==0.15.2
+#torch==2.0.1
+#torchvision==0.15.2