amazonaws-sp commited on
Commit
ed83b69
1 Parent(s): c220fd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(model, 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))
 
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))