Spaces:
Sleeping
Sleeping
amazonaws-sp
commited on
Commit
•
ed83b69
1
Parent(s):
c220fd6
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def generate(
|
|
59 |
if torch.cuda.is_available():
|
60 |
|
61 |
if not use_vae:
|
62 |
-
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
63 |
response = requests.get(url)
|
64 |
init_image = Image.open(BytesIO(response.content)).convert("RGB")
|
65 |
init_image = init_image.resize((768, 512))
|
|
|
59 |
if torch.cuda.is_available():
|
60 |
|
61 |
if not use_vae:
|
62 |
+
pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
|
63 |
response = requests.get(url)
|
64 |
init_image = Image.open(BytesIO(response.content)).convert("RGB")
|
65 |
init_image = init_image.resize((768, 512))
|