qinghuazhou commited on
Commit
9b292d2
·
1 Parent(s): f4ea072

updated demo

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -22,6 +22,7 @@ def load_editor(model_name='gpt2-xl'):
22
  model_name=model_name,
23
  hparams = hparams,
24
  layer = 13,
 
25
  edit_mode='in-place',
26
  verbose=True
27
  )
@@ -404,4 +405,4 @@ with gr.Blocks(theme=gr.themes.Soft(text_size="sm")) as demo:
404
 
405
 
406
  # launch demo
407
- demo.launch()
 
22
  model_name=model_name,
23
  hparams = hparams,
24
  layer = 13,
25
+ cache_path='/data/cache/',
26
  edit_mode='in-place',
27
  verbose=True
28
  )
 
405
 
406
 
407
  # launch demo
408
+ demo.launch()