update demo

edit_app.py

@@ -48,7 +48,9 @@ example_instructions = [
 model_id = "MudeHui/ip2p-warp-gpt4v"
 
 def main():
-    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None).to("cuda")
+    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
+    if torch.cuda.is_available():
+        pipe = pipe.to('cuda')
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     example_image = Image.open("imgs/example.jpg").convert("RGB")
 

requirements.txt

@@ -4,3 +4,4 @@ torchvision
 numpy
 transformers
 git+https://github.com/huggingface/diffusers
+accelerate