Zaixi commited on
Commit
c6eeef3
·
1 Parent(s): 33001d2
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -283,6 +283,7 @@ configs = parse_configs(
283
  )
284
  configs.load_checkpoint_path='./checkpoint.pt'
285
  download_infercence_cache(configs, model_version="v0.2.0")
 
286
  runner = InferenceRunner(configs)
287
  add_watermark = gr.Checkbox(label="Add Watermark", value=True)
288
  add_watermark1 = gr.Checkbox(label="Add Watermark", value=True)
 
283
  )
284
  configs.load_checkpoint_path='./checkpoint.pt'
285
  download_infercence_cache(configs, model_version="v0.2.0")
286
+ configs.use_deepspeed_evo_attention=False
287
  runner = InferenceRunner(configs)
288
  add_watermark = gr.Checkbox(label="Add Watermark", value=True)
289
  add_watermark1 = gr.Checkbox(label="Add Watermark", value=True)