vsrinivas commited on
Commit
b215fa0
1 Parent(s): 537a28c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def generate(prompt,steps,guidance,width,height):
16
  output = get_completion(prompt,params)
17
  return output
18
 
19
- pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
20
 
21
  with gr.Blocks() as demo:
22
  gr.Markdown("# Image Generation Demo & Test App by Srinivas")
 
16
  output = get_completion(prompt,params)
17
  return output
18
 
19
+ pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
20
 
21
  with gr.Blocks() as demo:
22
  gr.Markdown("# Image Generation Demo & Test App by Srinivas")