Hugging Face's logo

Spaces:
THUDM
/
CogView2
Runtime error

App Files Community
7
hysts HF staff commited on
Commit
5d7586b
·
1 Parent(s): 40139b3

Change max batch size to 8

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def set_example_text(example: list) -> list[dict]:
29
 
30
  def main():
31
  only_first_stage = True
32
- max_inference_batch_size = 4
33
  model = AppModel(max_inference_batch_size, only_first_stage)
34
 
35
  with gr.Blocks(css='style.css') as demo:
 
29
 
30
  def main():
31
  only_first_stage = True
32
+ max_inference_batch_size = 8
33
  model = AppModel(max_inference_batch_size, only_first_stage)
34
 
35
  with gr.Blocks(css='style.css') as demo: