wooyeolbaek commited on
Commit
a6279fd
1 Parent(s): 5f4f504

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -21,7 +21,10 @@ pipe = pipe.to(device)
21
 
22
 
23
  def inference(prompt):
24
- image = pipe(prompt).images[0]
 
 
 
25
  net_attn_maps = get_net_attn_map(image.size)
26
  net_attn_maps = resize_net_attn_map(net_attn_maps, image.size)
27
  net_attn_maps = return_net_attn_map(net_attn_maps, pipe.tokenizer, prompt)
 
21
 
22
 
23
  def inference(prompt):
24
+ image = pipe(
25
+ prompt,
26
+ num_inference_steps=15,
27
+ ).images[0]
28
  net_attn_maps = get_net_attn_map(image.size)
29
  net_attn_maps = resize_net_attn_map(net_attn_maps, image.size)
30
  net_attn_maps = return_net_attn_map(net_attn_maps, pipe.tokenizer, prompt)