Spaces:

alaa-lab
/

InstructCV

Sleeping

yulu2 commited on Aug 20, 2023

Commit

26b6838

•

1 Parent(s): 9c6e7dc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -37,7 +37,7 @@ model_id = "alaa-lab/InstructCV"
 def main():
     # pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None).to("cpu")
-    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, safety_checker=None).to("cpu")
     example_image = Image.open("imgs/example2.jpg").convert("RGB")

 def main():
     # pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None).to("cpu")
+    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, safety_checker=None).to("cuda")
     example_image = Image.open("imgs/example2.jpg").convert("RGB")