Spaces:

amazonaws-sp
/

zm

Sleeping

amazonaws-sp commited on Apr 7

Commit

9fdb4fb

•

1 Parent(s): 83d6fc0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -61,7 +61,7 @@ def generate(
     if torch.cuda.is_available():
         if use_img2img:
-            pipe = AutoPipelineForImage2Image.from_pretrained(model, torch_dtype=torch.float16)
             response = requests.get(url)
             init_image = Image.open(BytesIO(response.content)).convert("RGB")

     if torch.cuda.is_available():
         if use_img2img:
+            pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
             response = requests.get(url)
             init_image = Image.open(BytesIO(response.content)).convert("RGB")