jykoh commited on
Commit
868d95e
Β·
1 Parent(s): dda3162

Fix num_inference_steps

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -96,7 +96,7 @@ def generate_for_prompt(input_text, state, ret_scale_factor, num_words, temperat
96
  model_outputs = model.generate_for_images_and_texts(model_inputs,
97
  num_words=max(num_words, 1), ret_scale_factor=ret_scale_factor, top_p=top_p,
98
  temperature=temperature, max_num_rets=1,
99
- num_inference_steps=1)
100
  print('model_outputs', model_outputs, ret_scale_factor, flush=True)
101
 
102
  im_names = []
 
96
  model_outputs = model.generate_for_images_and_texts(model_inputs,
97
  num_words=max(num_words, 1), ret_scale_factor=ret_scale_factor, top_p=top_p,
98
  temperature=temperature, max_num_rets=1,
99
+ num_inference_steps=50)
100
  print('model_outputs', model_outputs, ret_scale_factor, flush=True)
101
 
102
  im_names = []